backend/app/search/services/autocomplete_schema.py

from typing import List, Dict

from search.models import Product, Category, Characteristic, UnitCharacteristic


def autocomplete_schema(val: str, exclude: List[Dict]):
    exclude = [dict(x) for x in exclude]
    name_exclude = [x["value"] for x in exclude if x["type"] == "Name"]
    category_exclude = [x["value"] for x in exclude if x["type"] == "Category"]
    schema = []
    if not category_exclude:
        schema.extend(
            [
                {
                    "coordinate": cat["name"].replace("ё", "е").lower().index(val.lower()),
                    "value": {"type": "Category", "value": cat["name"]},
                }
                for cat in Category.objects.filter(name__unaccent__icontains=val)[
                    :10
                ].values("name")
            ]
        )
    if not name_exclude:
        schema.extend(
            [
                {
                    "coordinate": product["name"].replace("ё", "е").lower().index(val.lower()),
                    "value": {
                        "type": "Name",
                        "value": product["name"],
                    },
                }
                for product in Product.objects.filter(name__unaccent__icontains=val)[
                    :30
                ].values("name")
            ]
        )
    schema.extend(
        [
            {
                "coordinate": char["value"].replace("ё", "е").lower().index(val.lower()),
                "value": {"type": char["name"], "value": char["value"]},
            }
            for char in Characteristic.objects.filter(value__unaccent__icontains=val)[
                :20
            ].values("name", "value")
        ]
    )
    schema.extend(
        [
            {
                "coordinate": char["name"].lower().replace("ё", "е").index(val.lower()),
                "value": {
                    "type": char["name"] + "_numeric",
                    "value": char["name"]
                }
            } for char in UnitCharacteristic.objects.filter(name__unaccent__icontains=val)[:20].values("name", "value")
        ]
    )
    return schema