django-haystack + whoosh + jieba 实现全文搜索

网站实现全文搜索,并对中文进行分词搜索

开发环境: Python3.7 Django3.2

需求:

网站内有商品、求购2个模块, 搜索栏输入 塑料玩具 时,希望优先搜索出匹配 塑料玩具 的信息, 并同时匹配出 塑料玩具等信息,按照匹配度排序。同时当输入 玩具塑料塑料玩巨 错别字时,同样能匹配到 塑料玩具 类的信息 。

匹配英文大小写

分析

1.django orm 的模糊匹配 icontains 表示 Mysql的like,不满足业务分词需求,并且随着数据量增大,消耗的资源和时间都会线性增长。
2.Elasticsearch 符合业务需求,但是部署时对服务器性能要求较高,网站的业务数据量预期不超过10万条。

选择 jieba 中文分词器 + whoosh 全文搜索引擎 + haystack搜索框架

代码实现
django   settings.py  配置

BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))

######################################
# haystack配置
######################################
HAYSTACK_CONNECTIONS = {
    'default': {
        'ENGINE': 'apps.framework.whoosh_cn_backend.WhooshEngine',
        'PATH': os.path.join(BASE_DIR, 'whoosh_index'),
    },
}
# 当添加、修改、删除数据时,自动生成索引
HAYSTACK_SIGNAL_PROCESSOR = 'haystack.signals.RealtimeSignalProcessor'
# 默认查询加载数据大小
HAYSTACK_ITERATOR_LOAD_PER_QUERY = 500

注意:
HAYSTACK_ITERATOR_LOAD_PER_QUERY 默认是10, 当索引库过大时,会导致频繁切词,影响查询速度。

jieba中文分词组件

import jieba
from whoosh.analysis import Tokenizer,Token



class ChineseTokenizer(Tokenizer):
    _instance = None

    def __new__(cls):
        if cls._instance is None:
            cls._instance = object.__new__(cls)
        return cls._instance
    
    def __call__(self, value, positions=False, chars=False,
                 keeporiginal=False, removestops=True,
                 start_pos=0, start_char=0, mode='', **kwargs):
        t = Token(positions, chars, removestops=removestops, mode=mode,
            **kwargs)
        seglist=jieba.cut(value,cut_all=True)

        for w in seglist:
            t.original = t.text = w
            t.boost = 1.0
            if positions:
                t.pos=start_pos+value.find(w)
            if chars:
                t.startchar=start_char+value.find(w)
                t.endchar=start_char+value.find(w)+len(w)

            yield t

复制 django-haystack 库中的 haystack.backends.backends.whoosh_backend.py 文件

引入上面的分词模块
from framework.chinese_analyzer import ChineseAnalyzerInstance

此时的匹配模式是分词  AND 匹配,如果想 OR 匹配,找到这行代码 改为:

self.parser = QueryParser(self.content_field_name, schema=self.schema, group=OrGroup)

接下来在创建索引
在模块中创建 search_index.py

from haystack import indexes
from business.models import Quotation

class QuotationIndex(indexes.SearchIndex, indexes.Indexable):
    text = indexes.CharField(document=True, use_template=True)

    def get_model(self):
        return Quotation

    def index_queryset(self, using=None):
        """返回要建立索引的数据查询集"""
        return self.get_model().objects.filter(is_delete=0)

在templates 下创建 txt
对应模块路径
templates/serach/indexes/business/quotation_text.txt

{{ object.title  }}

使用manager 命令生成 索引

python manage.py rebuild_index

最后,在view中使用吧

class VisitorQuotationView(viewsets.GenericViewSet):
    # 列表数据
    def list(self, request, *args, **kwargs):
        key = request.GET.get('key', "")
        all_results = SearchQuerySet().auto_query(key).models(Quotation)
        return Response({'data': all_results }, status=status.HTTP_200_OK)

到这里,一个简单的全文搜索就完成了。

当输入key是英文+数字时,whoosh将会使用全词匹配,不使用切词 + 模糊搜索

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 194,242评论 5 459
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 81,769评论 2 371
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 141,484评论 0 319
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,133评论 1 263
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,007评论 4 355
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,080评论 1 272
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,496评论 3 381
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,190评论 0 253
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,464评论 1 290
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,549评论 2 309
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,330评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,205评论 3 312
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,567评论 3 298
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 28,889评论 0 17
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,160评论 1 250
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,475评论 2 341
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 40,650评论 2 335

推荐阅读更多精彩内容