phzhik/kwork-poizonstore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

* WIP: annotate_commission_rub

Browse Source
This commit is contained in:
Phil Zhitnikov 2023-11-11 09:51:17 +04:00
parent 9a2a630465
commit 0e5c0ce16f
1 changed files with 29 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
store/models.py
View File

@ -277,15 +277,33 @@ class ChecklistQuerySet(models.QuerySet):
)
def annotate_commission_rub(self):
# FIXME: implement category commission in DB query
return self
default_commission = GlobalSettings.load().commission_rub
over_150k_commission = F('_price_rub') * settings.COMMISSION_OVER_150K
commission = GlobalSettings.load().commission_rub
return self.annotate(_commission_rub=Case(
When(GreaterThan(F("_price_rub"), 150_000), then=F("_price_rub") * settings.COMMISSION_OVER_150K),
default=commission,
output_field=DecimalField()
))
category_commission_is_zero_and_parent_present = (
(Q(category__commission__isnull=True) | Q(category__commission=0)) & Q(category__parent__isnull=False)
)
return self.annotate(
_category_commission_percent=Case(
When(category_commission_is_zero_and_parent_present, then=F('category__parent__commission')),
default=F('category__commission')
),
_category_commission=F('_category_commission_percent') * F('_price_rub') / 100,
_over_150k_commission=Case(
When(GreaterThan(F("_price_rub"), 150_000), then=over_150k_commission),
default=0,
output_field=DecimalField()
),
_commission_rub=Case(
When(price_snapshot_id__isnull=False, then=F('price_snapshot__commission_rub')),
default=Max(default_commission, F('_over_150k_commission'), F('_category_commission')),
output_field=DecimalField()
),
)
class PriceSnapshot(models.Model):
@ -485,10 +503,9 @@ class Checklist(models.Model):
@property
def commission_rub(self) -> Decimal:
# FIXME: implement category commission in DB query
# # Prefer annotated field
# if hasattr(self, '_commission_rub'):
# return self._commission_rub
# Prefer annotated field
if hasattr(self, '_commission_rub'):
return self._commission_rub
# Prefer saved value
if self.price_snapshot_id: