ElasticSearch и Python: проблема с функцией поиска - PullRequest
0 голосов
/ 22 октября 2018

Я впервые пытаюсь использовать ElasticSearch 6.4 с существующим веб-приложением, написанным на Python/Django.У меня есть некоторые проблемы, и я хотел бы понять, почему и как я могу решить эти проблемы.

###########

# Существующие: #

###########

В моем приложении можно загружать файлы документов (.pdfили .doc например).Затем в моем приложении есть функция поиска, которая позволяет выполнять поиск по документам, проиндексированным ElasticSearch, когда они загружены.

Название документа всегда пишется одинаково:

YEAR - DOC_TYPE - ORGANISATION - document_title.extension

Дляпример:

1970_ANNUAL_REPORT_APP-TEST_1342 - loremipsum.pdf

Функция поиска всегда выполняется среди doc_type = ANNUAL_REPORT.потому что есть несколько типов документов (ANNUAL_REPORT, OTHERS, ....).

##################

# Мое окружение: #

##################

Этоэто некоторые данные в соответствии с моей частью ElasticSearch.Я тоже изучаю команды ES.

$ curl -XGET http://127.0.0.1:9200/_cat/indices?v
health status index uuid                   pri rep docs.count docs.deleted store.size pri.store.size
yellow open   app  5T0HZTbmQU2-ZNJXlNb-zg   5   1        742            2    396.4kb        396.4kb

Итак, мой индекс app

Для приведенного выше примера, если я ищу этот документ: 1970_ANNUAL_REPORT_APP-TEST_1342 - loremipsum.pdf, у меня есть:

$ curl -XGET http://127.0.0.1:9200/app/annual-report/1343?pretty
{
  "_index" : "app",
  "_type" : "annual-report",
  "_id" : "1343",
  "_version" : 33,
  "found" : true,
  "_source" : {
    "attachment" : {
      "date" : "2010-03-04T12:08:00Z",
      "content_type" : "application/pdf",
      "author" : "manshanden",
      "language" : "et",
      "title" : "Microsoft Word - Test document Word.doc",
      "content" : "some text ...",
      "content_length" : 3926
    },
    "relative_path" : "app_docs/APP-TEST/1970_ANNUAL_REPORT_APP-TEST_1342.pdf",
    "title" : "1970_ANNUAL_REPORT_APP-TEST_1342 - loremipsum.pdf"
  }
}

Теперь, с моей поисковой частью в моем веб-приложении, я хотел бы найти этот документ с таким поиском: 1970.

def search_in_annual(self, q):
    try:
        response = self.es.search(
            index='app', doc_type='annual-report',
            q=q, _source_exclude=['data'], size=5000)
    except ConnectionError:
        return -1, None

    total = 0
    hits = []
    if response:
        for hit in response["hits"]["hits"]:
            hits.append({
                'id': hit['_id'],
                'title': hit['_source']['title'],
                'file': hit['_source']['relative_path'],
            })

        total = response["hits"]["total"]

    return total, hits

Но когда q=1970, результатis 0

Если я напишу:

response = self.es.search(
                index='app', doc_type='annual-report',
                q="q*", _source_exclude=['data'], size=5000)

Возвращает мой документ, но также много документов без 1970 внутри заголовка или содержимого документа.

#################

# Мой глобальный код: #

#################

Это глобальный класс, который управляет функциями индексации:

class EdqmES(object):
    host = 'localhost'
    port = 9200
    es = None

    def __init__(self, *args, **kwargs):
        self.host = kwargs.pop('host', self.host)
        self.port = kwargs.pop('port', self.port)

        # Connect to ElasticSearch server
        self.es = Elasticsearch([{
            'host': self.host,
            'port': self.port
        }])

    def __str__(self):
        return self.host + ':' + self.port

    @staticmethod
    def file_encode(filename):
        with open(filename, "rb") as f:
            return b64encode(f.read()).decode('utf-8')

    def create_pipeline(self):
        body = {
            "description": "Extract attachment information",
            "processors": [
                {"attachment": {
                    "field": "data",
                    "target_field": "attachment",
                    "indexed_chars": -1
                }},
                {"remove": {"field": "data"}}
            ]
        }
        self.es.index(
            index='_ingest',
            doc_type='pipeline',
            id='attachment',
            body=body
        )

    def index_document(self, doc, bulk=False):
        filename = doc.get_filename()

        try:
            data = self.file_encode(filename)
        except IOError:
            data = ''
            print('ERROR with ' + filename)
            # TODO: log error

        item_body = {
            '_id': doc.id,
            'data': data,
            'relative_path': str(doc.file),
            'title': doc.title,
        }

        if bulk:
            return item_body

        result1 = self.es.index(
            index='app', doc_type='annual-report',
            id=doc.id,
            pipeline='attachment',
            body=item_body,
            request_timeout=60
        )
        print(result1)
        return result1

    def index_annual_reports(self):
        list_docs = Document.objects.filter(category=Document.OPT_ANNUAL)

        print(list_docs.count())
        self.create_pipeline()

        bulk = []
        inserted = 0
        for doc in list_docs:
            inserted += 1
            bulk.append(self.index_document(doc, True))

            if inserted == 20:
                inserted = 0
                try:
                    print(helpers.bulk(self.es, bulk, index='app',
                                       doc_type='annual-report',
                                       pipeline='attachment',
                                       request_timeout=60))
                except BulkIndexError as err:
                    print(err)
                bulk = []

        if inserted:
            print(helpers.bulk(
                self.es, bulk, index='app',
                doc_type='annual-report',
                pipeline='attachment', request_timeout=60))

Мой документ проиндексированкогда он отправил спасибо форму Django с сигналом:

@receiver(signals.post_save, sender=Document, dispatch_uid='add_new_doc')
def add_document_handler(sender, instance=None, created=False, **kwargs):
    """ When a document is created index new annual report (only) with Elasticsearch and update conformity date if the
    document is a new declaration of conformity

    :param sender: Class which is concerned
    :type sender: the model class
    :param instance: Object which was just saved
    :type instance: model instance
    :param created: True for a creation, False for an update
    :type created: boolean
    :param kwargs: Additional parameter of the signal
    :type kwargs: dict
    """

    if not created:
        return

    # Index only annual reports
    elif instance.category == Document.OPT_ANNUAL:
        es = EdqmES()
        es.index_document(instance)

1 Ответ

0 голосов
/ 22 октября 2018

Это то, что я сделал, и похоже, что оно работает:

def search_in_annual(self, q):
    try:
        response = self.es.search(
            index='app', doc_type='annual-report', q=q, _source_exclude=['data'], size=5000)

        if response['hits']['total'] == 0:

            response = self.es.search(
                index='app', doc_type='annual-report',
                body={
                    "query":
                        {"prefix": {"title": q}},
                }, _source_exclude=['data'], size=5000)

    except ConnectionError:
        return -1, None

    total = 0
    hits = []
    if response:
        for hit in response["hits"]["hits"]:
            hits.append({
                'id': hit['_id'],
                'title': hit['_source']['title'],
                'file': hit['_source']['relative_path'],
            })

        total = response["hits"]["total"]
    return total, hits

Позволяет искать по заголовку, префиксу и содержимому, чтобы найти мой документ.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...