magnum-opus-tender-hack/backend
1
0
Fork 0
mirror of https://github.com/magnum-opus-tender-hack/backend.git synced 2024-11-25 10:43:44 +03:00
Code Issues Packages Projects Releases Wiki Activity

added translation check for Unknown

Browse Source
This commit is contained in:
Alexander Karpov 2022-10-22 11:20:24 +03:00
parent be3159dc74
commit 0ff8e5674b
1 changed files with 17 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
app/search/services/search.py
View File

@ -28,7 +28,7 @@ def process_unit_operation(unit: ProductUnitCharacteristic.objects, operation: s
return unit return unit
def process_search(data: List[dict], limit=10, offset=0) -> List[dict]: def process_search(data: List[dict], limit=5, offset=0) -> List[dict]:
prep_data = [] prep_data = []
prep_dict = {} prep_dict = {}
prep_dict_char_type = {} prep_dict_char_type = {}
@ -90,9 +90,9 @@ def process_search(data: List[dict], limit=10, offset=0) -> List[dict]:
characteristic__in=prep_dict_char_type[x["type"]], characteristic__in=prep_dict_char_type[x["type"]],
characteristic__value__unaccent__trigram_similar=val, characteristic__value__unaccent__trigram_similar=val,
) )
# ----------------------------------- apply filters on QuerySet -------------------------------------------------- #
for el, val in prep_dict.items(): for el, val in prep_dict.items():
prep_data.append({"type": el, "value": val}) prep_data.append({"type": el, "value": val})
# ----------------------------------- apply filters on QuerySet -------------------------------------------------- #
qs = Product.objects.filter() qs = Product.objects.filter()
for x in prep_data: for x in prep_data:
typ = x["type"] typ = x["type"]
@ -102,7 +102,21 @@ def process_search(data: List[dict], limit=10, offset=0) -> List[dict]:
elif typ == "Category": elif typ == "Category":
qs = qs.filter(category__name__unaccent__trigram_similar=val) qs = qs.filter(category__name__unaccent__trigram_similar=val)
elif typ == "Unknown": elif typ == "Unknown":
# add translate if val[0] in string.printable:
val = "".join(translate_en_ru(val))
else:
val = "".join(translate_ru_en(val))
type = get_hints(val)
if type == "Name":
qs = qs.filter(name__unaccent__trigram_similar=val)
elif type == "Category":
qs = qs.filter(category__name__unaccent__trigram_similar=val)
elif type == "Unknown":
continue
else:
qs = qs.filter(
characteristics__characteristic__name__unaccent__trigram_similar=val
)
continue continue
else: else:
if typ.startswith("*"): if typ.startswith("*"):