diff options
Diffstat (limited to 'pharos-dashboard/notification/signals.py')
| -rw-r--r-- | pharos-dashboard/notification/signals.py | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/pharos-dashboard/notification/signals.py b/pharos-dashboard/notification/signals.py new file mode 100644 index 0000000..c1f33ac --- /dev/null +++ b/pharos-dashboard/notification/signals.py @@ -0,0 +1,15 @@ +from django.db.models.signals import post_save +from django.dispatch import receiver + +from booking.models import Booking +from notification.models import BookingNotification + + +@receiver(post_save, sender=Booking) +def booking_notification_handler(sender, instance, **kwargs): + BookingNotification.objects.update_or_create( + booking=instance, type='booking_start', defaults={'submit_time': instance.start} + ) + BookingNotification.objects.update_or_create( + booking=instance, type='booking_end', defaults={'submit_time': instance.end} + )
\ No newline at end of file |