address comment

This commit is contained in:
Curtis Castrapel 2018-02-27 12:34:18 -08:00
parent 77a1600c13
commit 18c64fafe4
1 changed files with 8 additions and 4 deletions

View File

@ -45,10 +45,14 @@ class CertificateCreationSchema(CertificateSchema):
@post_load @post_load
def default_notification(self, data): def default_notification(self, data):
if not data['notifications']: if not data['notifications']:
notification_name = "DEFAULT_{0}".format(data['owner'].split('@')[0].upper()) data['notifications'] += notification_service.create_default_expiration_notifications(
data['notifications'] += notification_service.create_default_expiration_notifications(notification_name, [data['owner']]) "DEFAULT_{0}".format(data['owner'].split('@')[0].upper()),
notification_name = 'DEFAULT_SECURITY' [data['owner']],
data['notifications'] += notification_service.create_default_expiration_notifications(notification_name, current_app.config.get('LEMUR_SECURITY_TEAM_EMAIL')) )
data['notifications'] += notification_service.create_default_expiration_notifications(
'DEFAULT_SECURITY',
current_app.config.get('LEMUR_SECURITY_TEAM_EMAIL')
)
return data return data