Mirrored_Repos/InvenTree
1
0
Fork 0
mirror of https://github.com/inventree/InvenTree synced 2024-08-30 18:33:04 +00:00
Code Issues Packages Projects Releases Wiki 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
No known key found for this signature in database
GPG Key ID: F50EF5741D33E076
1 changed files with 9 additions and 0 deletions
Show Stats 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): def get_recipients(self):
raise NotImplementedError('The `get_recipients` method must be implemented!') raise NotImplementedError('The `get_recipients` method must be implemented!')
def setup(self):
return True
def cleanup(self): def cleanup(self):
return True return True
@ -136,6 +139,9 @@ def deliver_notification(cls: NotificationMethod, obj, entry_name: str, receiver
# Log start # Log start
logger.info(f"Notify users via '{method.method_name}' for notification '{entry_name}' for '{str(obj)}'") 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 = True
success_count = 0 success_count = 0
@ -154,6 +160,9 @@ def deliver_notification(cls: NotificationMethod, obj, entry_name: str, receiver
else: else:
raise NotImplementedError('No delivery method found') raise NotImplementedError('No delivery method found')
# Run cleanup for delivery method
method.cleanup()
# Log results # Log results
logger.info(f"Notified {success_count} users via '{method.method_name}' for notification '{entry_name}' for '{str(obj)}' successfully") logger.info(f"Notified {success_count} users via '{method.method_name}' for notification '{entry_name}' for '{str(obj)}' successfully")
if not success: if not success: