from elasticsearch import Elasticsearch
from elasticsearch_async import AsyncElasticsearch

from app.core import config


ES_HOST = f"{config.elasticsearch_server}:{config.elasticsearch_port}"


def get_elasticsearch() -> Elasticsearch:
    return Elasticsearch(hosts=[ES_HOST])

def get_async_elasticsearch(loop=None) -> AsyncElasticsearch:
    return AsyncElasticsearch(hosts=[ES_HOST], loop=loop)