Поиск данных и запрос Elasticsearch с Python

Вопрос задан: 8 месяцев назад Последняя активность: 8 месяцев назад
up 1 down

Я пытаюсь получить данные от Elasticsearch с помощью python. Мне удалось подключиться к источнику данных с помощью Python.

Подключиться к Elasticsearch

from elasticsearch import Elasticsearch

try:
  es = Elasticsearch(
      ['https:/movies-elk-prod-elastic-ssl.pac.com'],
      http_auth=('xxxxx', 'xxxx'),
      port=10202,
  )
  print ("Connected", es.info())
except Exception as ex:
  print ("Error:", ex)
es.info(pretty=True)

Входные данные

   {
                "_index": "movies",
                "_type": "movie",
                "_id": "4444169",
                "_score": null,
                "_source": {
                    "actor": [
                        "Josh Duhamel",
                        "Megan Fox",
                        "Shia LaBeouf"
                    ],
                    "director": [
                        "Michael Bay"
                    ],
                    "full_text": "When Sam Witwicky learns the truth about the ancient origins of the Transformers, he must accept his fate and merge with Optimus Prime and Bumblebee in their epic battle against the Decepticons, who are stronger back than ever and plan to destroy our world.!",
                    "title": "Transformers: Revenge of the Fallen",
                    "type": "movie"
                },
                "sort": [
                    1544310000000,
                    4.05
                ]
            },
            {
                "_index": "movies",
                "_type": "movie",
                "_id": "4051",
                "_score": null,
                "_source": {
                    "actor": [
                        "Josh Duhamel",
                        "Shia LaBeouf",
                        "Patrick Dempsey"
                    ],
                    "director": [
                        "Michael Bay"
                    ],
                    "full_text": "A mysterious event from the Earth's past threatens to unleash such a devastating war that the Transformers can not possibly save the planet on its own. Sam Witwicky and the Autobots fight the darkness to defend our world against the devastating evil of the Decepticons.",
                    "title": "Transformers: Dark of the Moon",
                    "type": "movie"
                },
                "sort": [
                    1544310000000,
                    4.03949
                ]
            },

Далее я хочу написать функцию Python, которая запрашивает эластичный поиск на основе actor например, если я напишу Josh Duhamel. Это должно дать мне все фильмы, содержащие Josh Duhamel.

В качестве следующего шага я хочу преобразовать данные в python фрейм данных. Я попробовал несколько вещей на основе функций, которые я нашел, но это не работает для меня (Ps. Я новичок вasticsearch, а также Python - :))

def search(uri, term):
    """Simple Elasticsearch Query"""
    query = json.dumps({
        "query": {
            "match": {
                "content": term
            }
        }
    })
    response = requests.get(uri, data=query)
    results = json.loads(response.text)
    return results

def format_results(results):
    """Print results nicely:
    doc_id) content
    """
    data = [doc for doc in results['hits']['hits']]
    for doc in data:
        print("%s) %s" % (doc['_id'], doc['_source']['content'])

def create_doc(uri, doc_data={}):
    """Create new document."""
    query = json.dumps(doc_data)
    response = requests.post(uri, data=query)
    print(response)

from elasticsearch import Elasticsearch

es = Elasticsearch()
res = es.search(index="test", doc_type="articles", body={"query": {"match": {"content": "fox"}}})
print("%d documents found" % res['hits']['total'])
for doc in res['hits']['hits']:
    print("%s) %s" % (doc['_id'], doc['_source']['content']))

Буду признателен за вашу помощь и понимание. заранее спасибо

1 ответ

up 0 down

Кусок кода, который вы разместили, ищет в ES поле с именем «контент», для того, чтобы эта работа достигла того, чего вы хотите достичь, вам нужно изменить часть контента на желаемое поле в вашем документе, например. "актер". То же самое касается других частей, где вы ищите поле «контент» в остальной части кода.

 "query": {
        "match": {
            "actor": term
        }
    }