from os import getenv

from elasticsearch import Elasticsearch


def get_client():
    return Elasticsearch(getenv("ELASTIC_URL"))


def add_document(index, id, doc):
    client = get_client()
    client.index(index=index, id=id, document=doc)


def refresh_index(index):
    client = get_client()
    client.indices.refresh(index=index)


def search(index):
    client = get_client()

    resp = client.search(
        index=index,
        query={
            # "query_string": {
            #   "query": "*puan*",
            #   "default_field": "data"
            # }
            "bool": {
                "must": [
                    {"query_string": {"query": "*new*", "default_field": "data"}},
                    # {
                    #     "match": {
                    #         "id": "1",
                    #     }
                    # }
                ]
            }
        },
        highlight={"fields": {"data": {}}},
    )
    print("Got %d hit(s):" % resp["hits"]["total"]["value"])

    for hit in resp["hits"]["hits"]:
        print(resp["hits"]["total"])
        print(resp["hits"])
        print(hit["_source"])
        print("%(name)s: %(data)s" % hit["_source"])