Django 中的 PostgreSQL 文本搜索未按预期工作

Question

我正在为我的 API 实现一个搜索功能，以便在请求时返回对象的属性。到目前为止，我已经尝试使用 全文搜索 ，它是可用的，但它有这些烦人的事情：必须正确拼写单词才能返回结果和部分搜索，例如“appl”而不是“苹果”，行不通。我也尝试过 Trigram Similarity，但对于长句子它失败了。如何在Django中实现既准确又模糊的搜索功能？

这行得通

这行不通

这是我的 views.py

from django.shortcuts import render
from rest_framework.response import Response
from rest_framework import status
from rest_framework.decorators import api_view
from .models import Object_Locations
from .serializers import Object_LocationsSerializer
from django.contrib.postgres.search import SearchVector, SearchQuery


def index(request):
   return render(request, 'main/base.html', {})

@api_view(['GET',])
def LocationsList(request):
    if request.method == 'GET':
        vector = SearchVector('name', 'desc', 'catergory')
        query = request.GET.get('search')

        if query:
            locations = Object_Locations.objects.annotate(search=vector,).filter(search=SearchQuery(query))
        else:
            locations = Object_Locations.objects.all()

        serializer = Object_LocationsSerializer(locations, many=True)
        return Response(serializer.data)

Answer 1

来自 SearchQuery 文档：

SearchQuery 将用户提供的术语翻译成搜索查询对象，数据库将其与搜索向量进行比较。默认情况下，用户提供的所有词都会通过词干算法，然后查找所有结果词的匹配项。

如果 search_type 是 'plain'（这是默认设置），则这些字词将被视为单独的关键字。如果 search_type 是 'phrase'，则这些字词将被视为单个短语。如果 search_type 是 'raw'，那么您可以提供带有术语和运算符的格式化搜索查询。

在上面的代码中，带有搜索短语 "a blac husky" 的 SearchQuery 被翻译成如下 SQL 代码：

... @@ plainto_tsquery('a blac husky') ...

因此，如果您想获得具有相似搜索词组的结果，可以组合查询词：

from django.contrib.postgres.search import SearchQuery, SearchVector
from django.shortcuts import render
from rest_framework import status
from rest_framework.decorators import api_view
from rest_framework.response import Response

from .models import Object_Locations
from .serializers import Object_LocationsSerializer


def index(request):
return render(request, 'main/base.html', {})


@api_view(['GET',])
def LocationsList(request):
    if request.method == 'GET':
        vector = SearchVector('name', 'desc', 'catergory')
        query_terms = request.GET.get('search')
        query_raw = ' | '.join(query_terms.split())
        query = SearchQuery(query_raw, search_type='raw')
        if query.value:
            locations = Object_Locations.objects.annotate(search=vector).filter(search=query)
        else:
            locations = Object_Locations.objects.all()
        serializer = Object_LocationsSerializer(locations, many=True)
        return Response(serializer.data)

在上面的代码中，带有搜索短语 "a blac husky" 的 SearchQuery 被翻译成如下 SQL 代码：

... @@ plainto_tsquery('a | blac | husky') ...

可以使用逻辑操作来组合 SearchQuery，如我链接的文档中所述。

如果您想进一步深入，可以阅读我写的关于该主题的文章：

“Full-Text Search in Django with PostgreSQL”