import smtplib
from email.mime.text import MIMEText
import configparser
import logging
class AlertManager:
def __init__(self, config):
self.logger = logging.getLogger('hydrobot')
self.from_addr = config.get("alerts", "from_address")
self.to_addr = config.get("alerts", "to_address")
self.mail_server = config.get("alerts", "mail_server")
self.username = config.get("alerts", "username")
self.password = config.get("alerts", "password")
def send_alert(self, message):
msg = MIMEText(message)
msg['Subject'] = "HydroBot Alert"
msg['From'] = self.from_addr
msg['To'] = self.to_addr
server = smtplib.SMTP(self.mail_server)
server.ehlo()
server.starttls()
server.login(self.username, self.password)
server.send_message(msg)
server.quit()