inventree/InvenTree
2
0
Fork 0
mirror of https://github.com/inventree/InvenTree.git synced 2025-10-30 20:55:42 +00:00
Code Activity

added setup function to notification method

Browse Source 
and added calls to deliver_method
This commit is contained in:
Matthias
2021-12-02 09:52:30 +01:00
parent 803936130c
commit 1b9ca41b46
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
InvenTree/common/notifications.py
View File

@@ -34,6 +34,9 @@ class NotificationMethod:
def get_recipients(self):
raise NotImplementedError('The `get_recipients` method must be implemented!')
def setup(self):
return True
def cleanup(self):
return True
@@ -136,6 +139,9 @@ def deliver_notification(cls: NotificationMethod, obj, entry_name: str, receiver
# Log start
logger.info(f"Notify users via '{method.method_name}' for notification '{entry_name}' for '{str(obj)}'")
# Run setup for delivery method
method.setup()
success = True
success_count = 0
@@ -154,6 +160,9 @@ def deliver_notification(cls: NotificationMethod, obj, entry_name: str, receiver
else:
raise NotImplementedError('No delivery method found')
# Run cleanup for delivery method
method.cleanup()
# Log results
logger.info(f"Notified {success_count} users via '{method.method_name}' for notification '{entry_name}' for '{str(obj)}' successfully")
if not success: