Python-Projekte/gctools/mail.py

from dataclasses import dataclass
import smtplib
import ssl
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart
from email.mime.application import MIMEApplication
from email.utils import make_msgid
import config


@dataclass
class Message:
    mailto: str
    subject: str
    html: str
    attachment: list


class mail:
    connected = False

    def __init__(self, cfg=None):
        self.cfg = cfg
        if cfg is None:
            self.cfg = config.config()
        self.mail_cfg = self.cfg.smtp
        self.reply_to = self.cfg.kunde.replace(' ', '-').lower() + '@global-cube.com'
        self.context = ssl.create_default_context()

    def __enter__(self):
        try:
            if self.mail_cfg.secure == 'ssl':
                self.mailserver = smtplib.SMTP_SSL(self.mail_cfg.server, self.mail_cfg.port, context=self.context)
            else:
                self.mailserver = smtplib.SMTP(self.mail_cfg.server, self.mail_cfg.port)

            self.mailserver.set_debuglevel(1)
            if self.mail_cfg.secure == 'tls':
                # self.context.options |= ssl.OP_NO_TLSv1_2 | ssl.OP_NO_TLSv1_3
                self.context.minimum_version = ssl.TLSVersion["TLSv1_2"]
                self.mailserver.ehlo()
                self.mailserver.starttls(context=self.context)

            print(self.mailserver.login(self.mail_cfg.username, self.mail_cfg.password))
            print(self.mailserver.noop())
            self.connected = True
        except smtplib.SMTPException as e:
            print(e)
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        if self.mailserver.connected:
            self.mailserver.quit()

    def send(self, mailto, subject, html, attachment=None):
        msg = self.message(mailto, subject, html, attachment)
        msg['Message-ID'] = make_msgid(domain=self.mail_cfg.server)
        res = {'id': msg['Message-ID'], 'mailto': mailto, 'status': 'OK'}

        try:
            result = self.mailserver.sendmail(self.mail_cfg.email, mailto, '\n' + msg.as_string())
            print(result)
        except smtplib.SMTPException as e:
            print(e)

    def message(self, mailto, subject, html, attachment):
        msg = MIMEMultipart('alternative')
        msg['Reply-To'] = self.reply_to
        msg['From'] = f'Global Cube <{self.mail_cfg.email}>'
        msg['To'] = mailto
        msg['Subject'] = subject

        # text = '\n'.join([x for x, _ in attachment])
        # msg.attach(MIMEText(text, 'plain'))
        msg.attach(MIMEText(html, 'html'))

        for name, filename in attachment or []:
            with open(filename, "rb") as f:
                part = MIMEApplication(f.read(), Name=name)
                part['Content-Disposition'] = f'attachment; filename="{name}"'
                msg.attach(part)
        return msg


if __name__ == '__main__':
    with mail() as m:
        m.send('robert.bedner@gmail.com', 'Test 123', ['C:\\GAPS\\Portal\\daten\\1_1_Taegliche_Erfolgskontrolle_1.pdf'])