From dcb6db186b016c5b21e99de9b7dbf64808e1e284 Mon Sep 17 00:00:00 2001 From: Mojtaba-z Date: Tue, 2 Dec 2025 17:01:35 +0330 Subject: [PATCH] fix - bug of stat type in quota stat service --- apps/product/services/quota_stat_service.py | 17 ++++++++--------- 1 file changed, 8 insertions(+), 9 deletions(-) diff --git a/apps/product/services/quota_stat_service.py b/apps/product/services/quota_stat_service.py index 4b332da..5b9535d 100644 --- a/apps/product/services/quota_stat_service.py +++ b/apps/product/services/quota_stat_service.py @@ -59,14 +59,11 @@ class QuotaStatsService: assigner_stat, created = OrganizationQuotaStats.objects.get_or_create( organization=assigner, quota=quota, + stat_type='distribution' ) - if created: - assigner_stat.stat_type = 'distribution' - assigner_stat.save() - if assigner_stat.stat_type == 'distribution': - assigner_stat.remaining_amount -= distribution.weight - assigner_stat.total_distributed += distribution.weight - assigner_stat.save() + assigner_stat.remaining_amount -= distribution.weight + assigner_stat.total_distributed += distribution.weight + assigner_stat.save() # ============== destination ================ assigned_stat, _ = OrganizationQuotaStats.objects.get_or_create( @@ -102,11 +99,13 @@ class QuotaStatsService: assigner_stat = OrganizationQuotaStats.objects.get( organization=assigner, - quota=quota + quota=quota, + stat_type='distribution' ) assigned_stat = OrganizationQuotaStats.objects.get( organization=assigned, - quota=quota + quota=quota, + stat_type='distribution' ) # create pricing attributes and broker pricing data