diff --git a/apps/notification/signals.py b/apps/notification/signals.py index 35fa917..dc7111b 100644 --- a/apps/notification/signals.py +++ b/apps/notification/signals.py @@ -1,10 +1,11 @@ +from django.db import transaction from django.db.models.signals import post_save from django.dispatch import receiver + +from apps.pos_device.models import Device from apps.warehouse.models import InventoryEntry from apps.warehouse.models import product_models -from apps.pos_device.models import Device from .models import Notification -from django.db import transaction @receiver(post_save, sender=InventoryEntry) @@ -29,6 +30,7 @@ def create_inventory_entry_notification(sender, instance, created, **kwargs): transaction.on_commit(_create_notification) + @receiver(post_save, sender=product_models.QuotaDistribution) # noqa def create_quota_distribution_notification(sender, instance, created, **kwargs): """ Create notification for organization for quota distribution creation """ @@ -38,7 +40,7 @@ def create_quota_distribution_notification(sender, instance, created, **kwargs): def _create_notification(): # create notification for organization on pos device - devices = Device.objects.filter(assignment__client__organization=instance.organization) + devices = Device.objects.filter(assignment__client__organization=instance.assigned_organization) for device in devices: Notification.objects.create( device=device,