57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
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, host, port, user, password, to):
|
|
self.host = host
|
|
self.port = port
|
|
self.user = user
|
|
self.password = password
|
|
self.to = to
|
|
|
|
@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
|
|
|
|
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()
|