fureunoir/schon
fureunoir/schon
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add auto_resolve_brand method to handle brand resolution

Browse source 
Introduced a new static method, auto_resolve_brand, to simplify and automate the resolution or creation of Brand objects based on a given brand name. Updated imports in core/vendors/__init__.py to include the Brand model. This enhances the system's ability to process brands effectively.
This commit is contained in:
Egor Pavlovich Gorbunov 2025-05-07 02:40:23 +03:00
parent 70cf2c23a1
commit 5d007c07a2
1 changed files with 10 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
core/vendors/__init__.py vendored
View file

@ -3,7 +3,7 @@ from contextlib import suppress
from math import ceil
from core.elasticsearch import process_query
from core.models import AttributeValue, Category, Product, Stock, Vendor
from core.models import AttributeValue, Category, Product, Stock, Vendor, Brand
from payments.errors import RatesError
from payments.utils import get_rates
@ -104,6 +104,15 @@ class AbstractVendor:
return Category.objects.get(uuid=uuid)
return Category.objects.get_or_create(name=category_name, is_active=False)[0]
@staticmethod
def auto_resolve_brand(brand_name: str):
if brand_name:
with suppress(KeyError):
uuid = process_query(brand_name)["brands"][0]["uuid"]
if uuid:
return Brand.objects.get(uuid=uuid)
return Brand.objects.get_or_create(name=brand_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()