我有两个型号Item
和Owner
:
class Item(models.Model):
name = models.CharField(max_length=255)
owner = models.ForeignKey(
Owner, related_name='owner_items',
on_delete=models.CASCADE,
)
is_featured = models.BooleanField(
choices=CHOICES, default=BOOL_NO
)
# ...
我在项目名称和描述字段中进行全文搜索(基于Django文档)。 PostgreSQL版本是10。
search_vectors = (
SearchVector('name', weight='A', config='english') +
SearchVector('description', weight='B', config='english')
)
terms = [SearchQuery(term) for term in keyword.split()]
search_query = functools.reduce(operator.or_, terms)
search_rank = SearchRank(
search_vectors, search_query, weights=[0.2, 0.4, 0.6, 1]
)
qs = Item.objects.all().annotate(
rank=search_rank
).filter(rank__gte=0.2).order_by('-rank')
我想在等式中引入Owner
,name
字段,如果is_featured
是true
,也会给排名提供一点提升。
在进行此搜索之前,我有Owner
实例模型。
您可以在name
中添加来自Owner
类的字段search_vector
,可能使用不同的weight
和相同的config
(它只是一个假设,因为您没有指定所有者模型定义或数据)。
使用is_featured
作为rank
的增强的方法可以是注释1(你可以使用其他值),如果它是True
然后将其添加到SearchRank
结果。
from django.db import models
from django.db.models import Case, Value, When
from django.contrib.postgres.search import (
SearchQuery, SearchRank, SearchVector,
)
search_vectors = (
SearchVector('name', weight='A', config='english') +
SearchVector('description', weight='B', config='english') +
SearchVector('owner__name', weight='C', config='english')
)
terms = [SearchQuery(term) for term in keyword.split()]
search_query = functools.reduce(operator.or_, terms)
search_rank = SearchRank(
search_vectors, search_query, weights=[0.2, 0.4, 0.6, 1]
)
qs = Item.objects.all().annotate(
featured_boost=Case(
When(is_featured=True, then=Value(1)),
default=Value(0),
output_field=models.IntegerField(),
)
).annotate(
rank=search_rank + featured_boost
).filter(rank__gte=0.2).order_by('-rank')