Working mail alerts + alert system tweaks

restic_compose_backup/alerts/smtp.py

@@ -1,14 +1,56 @@
+import os
+import smtplib
+import logging
+from email.mime.text import MIMEText
+
 from restic_compose_backup.alerts.base import BaseAlert
 
+logger = logging.getLogger(__name__)
+
 
 class SMTPAlert(BaseAlert):
     name = 'smtp'
 
-    def __init__(self):
-        pass
+    def __init__(self, host, port, user, password, to):
+        self.host = host
+        self.port = port
+        self.user = user
+        self.password = password
+        self.to = to
 
-    def create_from_config(self, config):
-        pass
+    @classmethod
+    def create_from_env(cls):
+        instance = cls(
+            os.environ.get('EMAIL_HOST'),
+            os.environ.get('EMAIL_PORT'),
+            os.environ.get('EMAIL_HOST_USER'),
+            os.environ.get('EMAIL_HOST_PASSWORD'),
+            (os.environ.get('EMAIL_SEND_TO') or "").split(','),
+        )
+        if instance.properly_configured:
+            return instance
 
-    def send(self, self, subject=None, attachment=None, alert_type=ALERT_ERROR):
-        pass
+        return None
+
+    @property
+    def properly_configured(self) -> bool:
+        return self.host and self.port and self.user and self.password and len(self.to) > 0
+
+    def send(self, subject: str = None, body: str = None, alert_type: str = 'INFO'):
+        # send_mail("Hello world!")
+        msg = MIMEText(body)
+        msg['Subject'] = f"[{alert_type}] {subject}"
+        msg['From'] = self.user
+        msg['To'] = ', '.join(self.to)
+
+        try:
+            logger.info("Connecting to %s port %s", self.host, self.port)
+            server = smtplib.SMTP_SSL(self.host, self.port)
+            server.ehlo()
+            server.login(self.user, self.password)
+            server.sendmail(self.user, self.to, msg.as_string())
+            logger.info('Email Sent')
+        except Exception as ex:
+            logger.error(ex)
+        finally:
+            server.close()