Add auto_resolve_category method to vendor logic

Introduced the auto_resolve_category method to streamline category resolution using Elasticsearch. This enhances category lookup efficiency and ensures seamless creation or retrieval of categories when necessary.
This commit is contained in:
Egor Pavlovich Gorbunov 2025-05-07 02:38:28 +03:00
parent 0e86c96ad3
commit 70cf2c23a1

View file

@ -2,6 +2,7 @@ import json
from contextlib import suppress
from math import ceil
from core.elasticsearch import process_query
from core.models import AttributeValue, Category, Product, Stock, Vendor
from payments.errors import RatesError
from payments.utils import get_rates
@ -94,6 +95,15 @@ class AbstractVendor:
# Default case: treat as a plain string.
return value, "string"
@staticmethod
def auto_resolve_category(category_name: str):
if category_name:
with suppress(KeyError):
uuid = process_query(category_name)["categories"][0]["uuid"]
if uuid:
return Category.objects.get(uuid=uuid)
return Category.objects.get_or_create(name=category_name, is_active=False)[0]
def resolve_price(self, original_price: int | float, vendor: Vendor = None, category: Category = None) -> float:
if not vendor:
vendor = self.get_vendor_instance()