Infra/lemur
5
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Condense sending notifications

Browse Source
This commit is contained in:
Jasmine Schladen 2020-11-13 20:10:21 -08:00
parent 79094e782d
commit db11f0c1b7
1 changed files with 12 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
lemur/notifications/messaging.py
View File

@ -153,29 +153,23 @@ def send_expiration_notifications(exclude):
).data
notification_data.append(cert_data)
email_recipients = security_email + [owner]
if notification.plugin.slug == "email-notification":
email_recipients = notification.plugin.get_recipients(notification.options, email_recipients)
if send_plugin_notification(
"expiration", notification_data, email_recipients, notification
):
success += len(email_recipients)
else:
failure += len(email_recipients)
email_recipients = notification.plugin.get_recipients(notification.options, security_email + [owner])
# Plugin will ONLY use the provided recipients if it's email; any other notification plugin ignores them
if send_plugin_notification(
"expiration", notification_data, email_recipients, notification
):
success += len(email_recipients)
else:
failure += len(email_recipients)
# If we're using an email plugin, we're done;
# if not, we also need to send an email notification to the security team and owner
if notification.plugin.slug != "email-notification":
if send_default_notification(
"expiration", notification_data, email_recipients, notification.options
):
success += len(email_recipients)
success = 1 + len(email_recipients)
else:
failure += len(email_recipients)
if send_plugin_notification(
"expiration", notification_data, [], notification
):
success += 1
else:
failure += 1
failure = 1 + len(email_recipients)
return success, failure