Egor fureunoir Gorbunov
ef553a94a4
Simplified field definitions and improved code consistency by consolidating and aligning styles. Added `prepare_*` methods to gracefully handle None values for fields. Enhanced multilingual support and streamlined query construction for better maintainability.
97 lines
No EOL
2.8 KiB
Python
97 lines
No EOL
2.8 KiB
Python
from django_elasticsearch_dsl import Document, fields
|
|
from django_elasticsearch_dsl.registries import registry
|
|
|
|
from core.elasticsearch import COMMON_ANALYSIS, ActiveOnlyMixin, _add_multilang_fields
|
|
from core.models import Brand, Category, Product
|
|
|
|
|
|
class _BaseDoc(ActiveOnlyMixin, Document):
|
|
name = fields.TextField(
|
|
attr="name",
|
|
analyzer="standard",
|
|
fields={
|
|
"raw": fields.KeywordField(ignore_above=256),
|
|
"ngram": fields.TextField(analyzer="name_ngram", search_analyzer="query_lc"),
|
|
"phonetic": fields.TextField(analyzer="name_phonetic"),
|
|
"auto": fields.TextField(analyzer="autocomplete", search_analyzer="autocomplete_search"),
|
|
},
|
|
)
|
|
description = fields.TextField(
|
|
attr="description",
|
|
analyzer="standard",
|
|
fields={
|
|
"raw": fields.KeywordField(ignore_above=256),
|
|
"ngram": fields.TextField(analyzer="name_ngram", search_analyzer="query_lc"),
|
|
"phonetic": fields.TextField(analyzer="name_phonetic"),
|
|
"auto": fields.TextField(analyzer="autocomplete", search_analyzer="autocomplete_search"),
|
|
},
|
|
)
|
|
|
|
class Index:
|
|
settings = {
|
|
"number_of_shards": 1,
|
|
"number_of_replicas": 0,
|
|
"analysis": COMMON_ANALYSIS,
|
|
"index": {"max_ngram_diff": 20},
|
|
}
|
|
|
|
def prepare_name(self, instance):
|
|
return getattr(instance, "name", "") or ""
|
|
|
|
def prepare_description(self, instance):
|
|
return getattr(instance, "description", "") or ""
|
|
|
|
|
|
class ProductDocument(_BaseDoc):
|
|
rating = fields.FloatField(attr="rating")
|
|
|
|
class Index(_BaseDoc.Index):
|
|
name = "products"
|
|
|
|
class Django:
|
|
model = Product
|
|
fields = ["uuid"]
|
|
|
|
|
|
_add_multilang_fields(ProductDocument)
|
|
registry.register_document(ProductDocument)
|
|
|
|
|
|
class CategoryDocument(_BaseDoc):
|
|
class Index(_BaseDoc.Index):
|
|
name = "categories"
|
|
|
|
class Django:
|
|
model = Category
|
|
fields = ["uuid"]
|
|
|
|
|
|
_add_multilang_fields(CategoryDocument)
|
|
registry.register_document(CategoryDocument)
|
|
|
|
|
|
class BrandDocument(ActiveOnlyMixin, Document):
|
|
name = fields.TextField(
|
|
attr="name",
|
|
analyzer="standard",
|
|
fields={
|
|
"raw": fields.KeywordField(ignore_above=256),
|
|
"ngram": fields.TextField(analyzer="name_ngram", search_analyzer="query_lc"),
|
|
"phonetic": fields.TextField(analyzer="name_phonetic"),
|
|
},
|
|
)
|
|
|
|
class Index:
|
|
name = "brands"
|
|
settings = {"number_of_shards": 1, "number_of_replicas": 0,
|
|
"analysis": COMMON_ANALYSIS, "index": {"max_ngram_diff": 18}}
|
|
|
|
class Django:
|
|
model = Brand
|
|
fields = ["uuid"]
|
|
|
|
def prepare_name(self, instance):
|
|
return getattr(instance, "name", "") or ""
|
|
|
|
|
|
registry.register_document(BrandDocument) |