add - quota stat services

2025-11-17 10:15:17 +03:30
parent 52cfc538d8
commit 2a34f1093a
2 changed files with 173 additions and 45 deletions
--- a/apps/product/services/quota_stat_service.py
+++ b/apps/product/services/quota_stat_service.py
@@ -0,0 +1,86 @@
+from apps.product.models import QuotaDistribution, OrganizationQuotaStats
+
+
+class QuotaStatsService:
+    @staticmethod
+    def apply_distribution(distribution: QuotaDistribution):
+        quota = distribution.quota
+
+        # origin org
+        assigner = distribution.assigner_organization
+        # destination org
+        assigned = distribution.assigned_organization
+        print(assigned)
+        # ================ origin ================
+        assigner_stat, _ = OrganizationQuotaStats.objects.get_or_create(
+            organization=assigner,
+            quota=quota
+        )
+
+        assigner_stat.remaining_amount -= distribution.weight
+        assigner_stat.total_distributed += distribution.weight
+        assigner_stat.save()
+
+        # ============== destination ================
+        assigned_stat, _ = OrganizationQuotaStats.objects.get_or_create(
+            organization=assigned,
+            quota=quota
+        )
+
+        assigned_stat.total_amount += distribution.weight
+        assigned_stat.remaining_amount += distribution.weight
+        assigned_stat.distributions.add(distribution)
+        assigned_stat.save()
+
+    @staticmethod
+    def update_distribution(distribution: QuotaDistribution, old_weight: int):
+        diff = distribution.weight - old_weight
+        if diff == 0:
+            return
+
+        quota = distribution.quota
+
+        assigner = distribution.assigner_organization
+        assigned = distribution.assigned_organization
+
+        assigner_stat = OrganizationQuotaStats.objects.get(
+            organization=assigner,
+            quota=quota
+        )
+        assigned_stat = OrganizationQuotaStats.objects.get(
+            organization=assigned,
+            quota=quota
+        )
+        # if diff > 0 it is added  to destination
+        assigner_stat.remaining_amount -= diff
+        assigner_stat.total_distributed += diff
+        assigner_stat.save()
+
+        assigned_stat.total_amount += diff
+        assigned_stat.remaining_amount += diff
+        print(assigned_stat.id)
+
+        assigned_stat.save()
+
+    @staticmethod
+    def delete_distribution(distribution: QuotaDistribution):
+        quota = distribution.quota
+
+        # origin
+        assigner_stat = OrganizationQuotaStats.objects.get(
+            organization=distribution.assigner_organization,
+            quota=quota
+        )
+        assigner_stat.remaining_amount += distribution.weight
+        assigner_stat.total_distributed -= distribution.weight
+        assigner_stat.save()
+
+        # destination
+        assigned_stat = OrganizationQuotaStats.objects.get(
+            organization=distribution.assigned_organization,
+            quota=quota
+        )
+        assigned_stat.total_amount -= distribution.weight
+        assigned_stat.remaining_amount -= distribution.weight
+        assigned_stat.distributions.remove(distribution)
+        assigned_stat.save()