Add notification on update of lending instance

webkom · Mar 12, 2024 · 37e3a49 · 37e3a49
1 parent e549314
commit 37e3a49
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/lego/apps/lending/managers.py b/lego/apps/lending/managers.py
@@ -26,6 +26,19 @@ def create(self, *args, **kwargs):
             notification.notify()
 
         return lending_instance
+
+    def update(self, *args, **kwargs):
+        from lego.apps.lending.notifications import LendingInstanceNotification
+
+        lending_instance = super().update(*args, **kwargs)
+        notification = LendingInstanceNotification(
+            lending_instance=lending_instance,
+            user=lending_instance.created_by,
+        )
+        notification.notify()
+
+        return lending_instance
+
 
     def get_queryset(self):
         return super().get_queryset().select_related("created_by")