37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
from django.db.models import Sum
|
|
from django.db.models.signals import post_save, post_delete
|
|
from django.dispatch import receiver
|
|
from .models import QuotaDistribution, Quota, Product
|
|
from apps.warehouse.models import InventoryQuotaSaleTransaction
|
|
|
|
|
|
def recalculate_remaining_amount(quota):
|
|
total_distributed = quota.distributions_assigned.aggregate(
|
|
total=Sum('weight')
|
|
)['total'] or 0
|
|
|
|
quota.remaining_weight = quota.quota_weight - total_distributed
|
|
quota.quota_distributed = total_distributed
|
|
quota.save(update_fields=["remaining_weight", "quota_distributed"])
|
|
|
|
|
|
@receiver(post_save, sender=QuotaDistribution)
|
|
@receiver(post_delete, sender=QuotaDistribution)
|
|
def update_quota_remaining(sender, instance, **kwargs):
|
|
recalculate_remaining_amount(instance.quota)
|
|
|
|
|
|
def update_product_stats(instance: Product):
|
|
pass
|
|
|
|
|
|
def update_quota_stats(instance: Quota):
|
|
pass
|
|
|
|
|
|
@receiver([post_save, post_delete], sender=QuotaDistribution)
|
|
@receiver([post_save, post_delete], sender=InventoryQuotaSaleTransaction)
|
|
def update_stats_on_change(sender, instance, **kwargs):
|
|
update_product_stats(instance)
|
|
update_quota_stats(instance)
|