diff --git a/apps/product/services/quota_stat_service.py b/apps/product/services/quota_stat_service.py index 63b7f5d..4b332da 100644 --- a/apps/product/services/quota_stat_service.py +++ b/apps/product/services/quota_stat_service.py @@ -59,11 +59,14 @@ class QuotaStatsService: assigner_stat, created = OrganizationQuotaStats.objects.get_or_create( organization=assigner, quota=quota, - stat_type='distribution' ) - assigner_stat.remaining_amount -= distribution.weight - assigner_stat.total_distributed += distribution.weight - assigner_stat.save() + 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() # ============== destination ================ assigned_stat, _ = OrganizationQuotaStats.objects.get_or_create(