RasadDam_Backend/apps/product/services/quota_stat_service.py

from apps.product.models import QuotaDistribution, OrganizationQuotaStats
from apps.product.validators.quota_stats_validator import QuotaStatsValidator


class QuotaStatsService:
    @staticmethod
    def apply_distribution(distribution: QuotaDistribution):
        quota = distribution.quota
        print("ssss")
        # origin org
        assigner = distribution.assigner_organization
        # destination org
        assigned = distribution.assigned_organization

        QuotaStatsValidator.validate_assigner_has_enough(
            assigner_org=assigner,
            quota=quota,
            amount=distribution.weight
        )

        # ================ origin ================
        assigner_stat, created = OrganizationQuotaStats.objects.get_or_create(
            organization=assigner,
            quota=quota,
        )
        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(
            organization=assigned,
            quota=quota,
            stat_type='distribution'
        )

        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

        QuotaStatsValidator.validate_distribution_update(
            assigner_org=assigner,
            assigned_org=assigned,
            quota=quota,
            old_amount=old_weight,
            new_amount=distribution.weight
        )

        assigner_stat = OrganizationQuotaStats.objects.get(
            organization=assigner,
            quota=quota
        )
        assigned_stat = OrganizationQuotaStats.objects.get(
            organization=assigned,
            quota=quota
        )
        if assigner_stat.stat_type == 'distribution':
            # 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
        assigned_stat.save()

    @staticmethod
    def delete_distribution(distribution: QuotaDistribution):
        quota = distribution.quota

        # origin
        assigner_stat = OrganizationQuotaStats.objects.get(
            organization=distribution.assigner_organization,
            quota=quota
        )
        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(
            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()