获取 django-elasticsearch-dsl 中所有文档的列表 - Python

如果你正在使用 Django 和 Elasticsearch 进行开发，那么使用 django-elasticsearch-dsl 库可以让你更方便的使用 Elasticsearch。

安装 django-elasticsearch-dsl

首先需要安装 django-elasticsearch-dsl 。通过 pip 安装：

pip install django-elasticsearch-dsl

定义模型

为了将 Django 模型映射到 Elasticsearch，需要定义一个 Elasticsearch 模型。

假设我们有一个名为 Article 的 Django 模型：

from django.db import models

class Article(models.Model):
    title = models.CharField(max_length=255)
    content = models.TextField()
    published_date = models.DateTimeField()

可以使用 django-elasticsearch-dsl 中提供的 Document 类将其映射到 Elasticsearch：

from django_elasticsearch_dsl import Document
from django_elasticsearch_dsl.registries import registry
from .models import Article

@registry.register_document
class ArticleDocument(Document):
    class Index:
        name = 'articles'
        settings = {
          "number_of_shards": 1,
        }

    class Django:
        model = Article

        fields = [
            'title',
            'content',
            'published_date',
        ]

其中，name 属性指定了 Elasticsearch 索引的名称。settings 属性用于设置 Elasticsearch 索引的配置。

检索所有文档

获取 django-elasticsearch-dsl 中所有文档的列表可以使用 Elasticsearch 的 Scroll API 。

以下是获取所有文章的示例代码：

from elasticsearch_dsl import Search
from elasticsearch.helpers import scan
from .documents import ArticleDocument

def get_all_articles():
    s = Search(index='articles')
    s = s.source(['title', 'content', 'published_date'])

    response = scan(
        s,
        index='articles',
        doc_type='ArticleDocument',
    )

    articles = [doc for doc in response]

    return articles

首先需要创建一个 Search 对象，指定 Elasticsearch 索引的名称。使用 source 方法指定返回的字段。

然后，使用 elasticsearch.helpers.scan 函数执行查询，返回的是一个迭代器，迭代器返回的是查询结果中每个文档的原始表示。在迭代结果中，可以使用 elasticsearch_dsl.Document 类从原始表示中获取文档对象的实例。

最后将所有文档对象的实例组成一个列表返回。

总结

本文介绍了如何使用 django-elasticsearch-dsl 获取所有文档的列表。需要安装 django-elasticsearch-dsl 库，并定义 Elasticsearch 模型将 Django 模型映射到 Elasticsearch。使用 Elasticsearch 的 Scroll API 检索所有文档。