如何在Django項(xiàng)目中使用Elasticsearch搜索引擎

本篇文章為大家展示了如何在Django項(xiàng)目中使用Elasticsearch搜索引擎，內(nèi)容簡明扼要并且容易理解，絕對(duì)能使你眼前一亮，通過這篇文章的詳細(xì)介紹希望你能有所收獲。

1.使用Docker安裝Elasticsearch及其擴(kuò)展

獲取鏡像，可以通過網(wǎng)絡(luò)pull

sudo docker image pull delron/elasticsearch-ik:2.4.6-1.0

或者加載鏡像文件

sudo docker load -i elasticsearch-ik-2.4.6_docker.tar

修改elasticsearch的配置文件 elasticsearc-2.4.6/config/elasticsearch.yml第54行，更改ip地址為本機(jī)ip地址

network.host: 127.0.0.1

創(chuàng)建docker容器運(yùn)行

sudo docker run -dti --network=host --name=elasticsearch -v /home/python/elasticsearch-2.4.6/config:/usr/share/elasticsearch/config delron/elasticsearch-ik:2.4.6-1.0

2. 使用haystack對(duì)接Elasticsearch

1）安裝

pip install drf-haystack
pip install elasticsearch==2.4.1

drf-haystack是為了在REST framework中使用haystack而進(jìn)行的封裝（如果在Django中使用haystack，則安裝django-haystack即可）。

2）注冊應(yīng)用

INSTALLED_APPS = [
 ...
 'haystack',
 ...
]

3）配置

在配置文件中配置haystack使用的搜索引擎后端

# Haystack
HAYSTACK_CONNECTIONS = {
 'default': {
  'ENGINE': 'haystack.backends.elasticsearch_backend.ElasticsearchSearchEngine',
  'URL': 'http://127.0.0.1:9200/', # 此處為elasticsearch運(yùn)行的服務(wù)器ip地址，端口號(hào)固定為9200
  'INDEX_NAME': 'meiduo', # 指定elasticsearch建立的索引庫的名稱
 },
}

# 當(dāng)添加、修改、刪除數(shù)據(jù)時(shí)，自動(dòng)生成索引
HAYSTACK_SIGNAL_PROCESSOR = 'haystack.signals.RealtimeSignalProcessor'

4）創(chuàng)建索引類

指明讓搜索引擎對(duì)哪些字段建立索引

goods應(yīng)用中新建search_indexes.py文件，用于存放索引類(文件名固定)

from haystack import indexes

from .models import SKU


class SKUIndex(indexes.SearchIndex, indexes.Indexable):
 """
 SKU索引數(shù)據(jù)模型類
 """
 text = indexes.CharField(document=True, use_template=True)

 def get_model(self):
  """返回建立索引的模型類"""
  return SKU

 def index_queryset(self, using=None):
  """返回要建立索引的數(shù)據(jù)查詢集"""
  return self.get_model().objects.filter(is_launched=True)

document=True 表名該字段是主要進(jìn)行關(guān)鍵字查詢的字段

use_template=True 表示通過模板來指明索引值由哪些模型類字段組成

5）在templates目錄中創(chuàng)建text字段使用的模板文件

在templates/search/indexes/goods/sku_text.txt文件中定義

{{ object.name }}
{{ object.caption }}
{{ object.id }}

路徑固定: templates/search/indexes/建立索引的應(yīng)用名/文件名.txt

通過sku的name、caption、id來進(jìn)行關(guān)鍵字索引查詢

6）手動(dòng)生成初始索引

python manage.py rebuild_index

7）創(chuàng)建序列化器

在goods/serializers.py中創(chuàng)建haystack序列化器

from drf_haystack.serializers import HaystackSerializer


class SKUIndexSerializer(HaystackSerializer):
 """
 SKU索引結(jié)果數(shù)據(jù)序列化器
 """
 object = SKUSerializer(read_only=True)

 class Meta:
  index_classes = [SKUIndex]
  fields = ('text', 'object')

說明：

使用SKUIndexSerializer序列化器用來檢查前端傳入的參數(shù)text，并且檢索出數(shù)據(jù)后再使用這個(gè)序列化器返回給前端；

SKUIndexSerializer序列化器中的object字段是用來向前端返回?cái)?shù)據(jù)時(shí)序列化的字段。

8）創(chuàng)建視圖

在goods/views.py中創(chuàng)建視圖

from drf_haystack.viewsets import HaystackViewSet

class SKUSearchViewSet(HaystackViewSet):
 """
 SKU搜索
 """
 index_models = [SKU]

 serializer_class = SKUIndexSerializer

9）定義路由

通過REST framework的router來定義路由

router = DefaultRouter()
router.register('skus/search', views.SKUSearchViewSet, base_name='skus_search')
urlpatterns += router.urls

bug說明：

如果在配置完haystack并啟動(dòng)程序后，出現(xiàn)如下異常，是因?yàn)閐rf-haystack還沒有適配最新版本的REST framework框架

importError: cannot import name '_get_count'

可以通過修改REST framework框架代碼，補(bǔ)充_get_count函數(shù)定義即可

文件路徑 虛擬環(huán)境下的 lib/python3.6/site-packages/rest_framework/pagination.py

def _get_count(queryset):
 """
 Determine an object count, supporting either querysets or regular lists.
 """
 try:
  return queryset.count()
 except (AttributeError, TypeError):
  return len(queryset)

上述內(nèi)容就是如何在Django項(xiàng)目中使用Elasticsearch搜索引擎，你們學(xué)到知識(shí)或技能了嗎？如果還想學(xué)到更多技能或者豐富自己的知識(shí)儲(chǔ)備，歡迎關(guān)注億速云行業(yè)資訊頻道。