34 lines
1.4 KiB
Python
34 lines
1.4 KiB
Python
from django.contrib.postgres.aggregates import StringAgg
|
|
from django.contrib.postgres.search import (
|
|
SearchQuery, SearchRank, SearchVector, TrigramSimilarity,
|
|
)
|
|
from django.db import models
|
|
|
|
|
|
# TODO add search highlighting
|
|
# TODO add language support
|
|
# TODO add schedule index rebuild
|
|
# TODO add admin function to rebuild index
|
|
class RecipeSearchManager(models.Manager):
|
|
|
|
def search(self, search_text, space):
|
|
search_query = SearchQuery(
|
|
search_text, config='english'
|
|
)
|
|
search_vectors = (
|
|
SearchVector('search_vector')
|
|
+ SearchVector(StringAgg('steps__ingredients__food__name', delimiter=' '), weight='B', config='english')
|
|
+ SearchVector(StringAgg('keywords__name', delimiter=' '), weight='B', config='english'))
|
|
search_rank = SearchRank(search_vectors, search_query)
|
|
# the results from trigram were really, really bad
|
|
# trigram = (
|
|
# TrigramSimilarity('name', search_text)
|
|
# + TrigramSimilarity('description', search_text)
|
|
# + TrigramSimilarity('steps__ingredients__food__name', search_text)
|
|
# + TrigramSimilarity('keywords__name', search_text))
|
|
return (
|
|
self.get_queryset()
|
|
.annotate(search=search_vectors, rank=search_rank)
|
|
.filter(search=search_query)
|
|
.order_by('-rank'))
|