AIL-framework/bin/NotificationHelper.py

81 lines
2.6 KiB
Python
Raw Normal View History

2018-05-04 13:53:29 +02:00
#!/usr/bin/env python3
# -*-coding:UTF-8 -*
import os
import sys
2018-10-01 15:56:48 +02:00
import argparse
import traceback
import smtplib
from pubsublogger import publisher
2018-04-16 14:50:04 +02:00
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText
sys.path.append(os.path.join(os.environ['AIL_BIN'], 'lib/'))
import ConfigLoader
"""
This module allows the global configuration and management of notification settings and methods.
"""
config_loader = ConfigLoader.ConfigLoader()
publisher.port = 6380
publisher.channel = "Script"
2018-07-16 15:51:37 +02:00
def sendEmailNotification(recipient, alert_name, content):
2018-04-16 14:50:04 +02:00
sender = config_loader.get_config_str("Notifications", "sender")
sender_user = config_loader.get_config_str("Notifications", "sender_user")
sender_host = config_loader.get_config_str("Notifications", "sender_host")
sender_port = config_loader.get_config_int("Notifications", "sender_port")
sender_pw = config_loader.get_config_str("Notifications", "sender_pw")
2018-11-05 14:30:03 +01:00
if sender_pw == 'None':
2018-11-05 14:20:12 +01:00
sender_pw = None
2018-03-30 11:35:37 +02:00
2018-04-04 09:41:13 +02:00
# raise an exception if any of these is None
if (sender is None or
sender_host is None or
sender_port is None
):
raise Exception('SMTP configuration (host, port, sender) is missing or incomplete!')
2018-03-30 11:35:37 +02:00
2018-04-04 09:41:13 +02:00
try:
if sender_pw is not None:
2018-07-16 15:51:37 +02:00
try:
smtp_server = smtplib.SMTP(sender_host, sender_port)
smtp_server.starttls()
except smtplib.SMTPNotSupportedError:
print("The server does not support the STARTTLS extension.")
smtp_server = smtplib.SMTP_SSL(sender_host, sender_port)
2018-04-04 09:41:13 +02:00
smtp_server.ehlo()
if sender_user is not None:
smtp_server.login(sender_user, sender_pw)
else:
smtp_server.login(sender, sender_pw)
2018-04-04 09:41:13 +02:00
else:
smtp_server = smtplib.SMTP(sender_host, sender_port)
2018-04-16 14:50:04 +02:00
2018-04-04 09:41:13 +02:00
mime_msg = MIMEMultipart()
mime_msg['From'] = sender
mime_msg['To'] = recipient
2018-10-01 15:56:48 +02:00
mime_msg['Subject'] = "AIL Framework " + alert_name + " Alert"
2018-04-16 14:50:04 +02:00
2018-07-16 15:51:37 +02:00
body = content
2018-04-04 09:41:13 +02:00
mime_msg.attach(MIMEText(body, 'plain'))
2018-04-16 14:50:04 +02:00
2018-04-04 09:41:13 +02:00
smtp_server.sendmail(sender, recipient, mime_msg.as_string())
smtp_server.quit()
2018-10-01 15:56:48 +02:00
print('Send notification ' + alert_name + ' to '+recipient)
2018-04-16 14:50:04 +02:00
except Exception as err:
traceback.print_tb(err.__traceback__)
publisher.warning(err)
2018-10-01 15:56:48 +02:00
if __name__ == '__main__':
parser = argparse.ArgumentParser(description='Test notification sender.')
parser.add_argument("addr", help="Test mail 'to' address")
args = parser.parse_args()
sendEmailNotification(args.addr, '_mail test_', 'Success.')