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