sanspie/akarpov
1
1
Fork 0
mirror of https://github.com/Alexander-D-Karpov/akarpov synced 2024-11-21 19:46:38 +03:00
Code Issues Packages Projects Releases Wiki Activity

updated music search

Browse Source
This commit is contained in:
Alexander Karpov 2024-04-22 14:17:11 +03:00
parent 4a2b86509e
commit 7e188865cc
2 changed files with 15 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
akarpov/music/documents.py
View File

@ -35,7 +35,8 @@ class SongDocument(Document):
name = fields.TextField(
attr="name",
fields={
"raw": fields.KeywordField(normalizer="lowercase"),
"raw": fields.KeywordField(),
"exact": fields.KeywordField(normalizer="lowercase"),
},
)
suggest = fields.CompletionField()

22
akarpov/music/services/search.py
View File

@ -12,8 +12,11 @@ def search_song(query):
search_query = ES_Q(
"bool",
should=[
ES_Q("match", name=query),
ES_Q("match", name__russian=query),
ES_Q("match_phrase", name__exact=query, boost=3), # Boost for exact matches
ES_Q(
"match_phrase", name__raw=query.lower(), boost=2.5
), # Case-insensitive exact match
ES_Q("match", name__russian=query), # Match in Russian field
ES_Q(
"multi_match",
query=query,
@ -24,7 +27,7 @@ def search_song(query):
"authors.name.raw^3",
"album.name^3",
"album.name.raw^3",
"name.raw^2",
"meta.*",
],
type="best_fields",
fuzziness="AUTO",
@ -38,8 +41,8 @@ def search_song(query):
fields=["authors.name", "authors.name.raw"],
fuzziness="AUTO",
),
boost=2,
),
ES_Q("wildcard", name__raw=f"*{query.lower()}*"),
ES_Q(
"nested",
path="album",
@ -49,13 +52,14 @@ def search_song(query):
fields=["album.name", "album.name.raw"],
fuzziness="AUTO",
),
boost=2,
),
ES_Q(
"nested",
path="album",
query=ES_Q("wildcard", album__name__raw=f"*{query.lower()}*"),
),
ES_Q("wildcard", meta__raw=f"*{query.lower()}*"),
"wildcard", name__raw=f"*{query.lower()}*"
), # Wildcard search for partial matches
ES_Q(
"wildcard", meta__raw=f"*{query.lower()}*"
), # Wildcard search in meta fields
],
minimum_should_match=1,
)