from elasticsearch import Elasticsearch

from delphi_es_utils.settings import ELASTICSEARCH_HOST, ELASTICSEARCH_PORT


def get_es_client(host=ELASTICSEARCH_HOST, port=ELASTICSEARCH_PORT):
    """Wrapper to create an (additional) Elasticsearch connection.

    Args:
      host(str): (optional) Elasticsearch instance hostname
      port(int or str): (optional) Elasticsearch instance port

    Returns:
      Elasticsearch: Elasticsearch client connection

    See Also:
        :class:`elasticsearch.transport.Transport` for additional list of available params
    """

    return Elasticsearch(
        f'https://{host}:{port}',
        timeout=120,
        retry_on_timeout=True,
        max_retries=3,  # default is 3
    )