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

refactor method name

Browse Source
This commit is contained in:
Matthias 2021-12-04 00:58:13 +01:00
parent 33b7d9b845
commit 0f3a78c7be
No known key found for this signature in database
GPG Key ID: F50EF5741D33E076
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
InvenTree/common/notifications.py
View File

@ -17,7 +17,7 @@ logger = logging.getLogger('inventree')
# region notification classes # region notification classes
# region base classes # region base classes
class NotificationMethod: class NotificationMethod:
method_name = '' METHOD_NAME = ''
def __init__(self, obj, entry_name, receivers) -> None: def __init__(self, obj, entry_name, receivers) -> None:
# Check if a sending fnc is defined # Check if a sending fnc is defined
@ -55,7 +55,7 @@ class BulkNotificationMethod(NotificationMethod):
# region implementations # region implementations
class EmailNotification(BulkNotificationMethod): class EmailNotification(BulkNotificationMethod):
method_name = 'mail' METHOD_NAME = 'mail'
def get_recipients(self): def get_recipients(self):
return EmailAddress.objects.filter( return EmailAddress.objects.filter(
@ -124,7 +124,7 @@ def trigger_notifaction(obj, entry_name=None, obj_ref='pk', receivers=None, rece
print(delivery_methods) print(delivery_methods)
for method in [a for a in delivery_methods if a not in [SingleNotificationMethod, BulkNotificationMethod]]: for method in [a for a in delivery_methods if a not in [SingleNotificationMethod, BulkNotificationMethod]]:
logger.info(f"Triggering method '{method.method_name}'") logger.info(f"Triggering method '{method.METHOD_NAME}'")
try: try:
deliver_notification(method, obj, entry_name, receivers, notification_context) deliver_notification(method, obj, entry_name, receivers, notification_context)
except NotImplementedError as error: except NotImplementedError as error:
@ -146,7 +146,7 @@ def deliver_notification(cls: NotificationMethod, obj, entry_name: str, receiver
if method.recipients and method.recipients.count() > 0: if method.recipients and method.recipients.count() > 0:
# 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 # Run setup for delivery method
method.setup() method.setup()
@ -173,6 +173,6 @@ def deliver_notification(cls: NotificationMethod, obj, entry_name: str, receiver
method.cleanup() 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:
logger.info("There were some problems") logger.info("There were some problems")