Coverage for apps/accounting/signals.py: 70%

1from django.db.models import Sum, signals 

2from django.dispatch import receiver 

3 

4from apps.campaigns.models import CampaignClaim 

5from apps.rewards.models import RewardClaim 

6 

7from .models import Accounting 

8 

9 

10@receiver(signals.pre_save, sender=Accounting) 

11def update_balance_on_account_save(sender, instance, **kwargs): 

12 campaign_balance = ( 

13 CampaignClaim.objects.filter(fan=instance.fan, campaign__organization=instance.organization, is_valid=True) 

14 .aggregate(campaign_balance=Sum("campaign__value")) 

15 .get("campaign_balance") 

16 or 0 

17 ) 

18 reward_balance = ( 

19 RewardClaim.objects.filter(fan=instance.fan, reward__organization=instance.organization) 

20 .aggregate(reward_balance=Sum("reward__points_required")) 

21 .get("reward_balance") 

22 or 0 

23 ) 

24 instance.balance = campaign_balance - reward_balance