from email.message import EmailMessage
from email.utils import formataddr
import smtplib


class Mail:
    def __init__(self, server, port, username, password, nickname=None):
        self.host = server
        self.port = int(port)
        self.user = username
        self.password = password
        self.nickname = nickname

    def _get_server(self):
        if self.port == 465:
            server = smtplib.SMTP_SSL(self.host, self.port)
        else:
            server = smtplib.SMTP(self.host, self.port)
        server.login(self.user, self.password)
        return server

    def send_text_mail(self, to_addrs, subject, body):
        if isinstance(to_addrs, (list, tuple)):
            to_addrs = ', '.join(to_addrs)
        server = self._get_server()
        msg = EmailMessage()
        msg.set_content(body)
        msg['Subject'] = subject
        msg['From'] = formataddr((self.nickname, self.user)) if self.nickname else self.user
        msg['To'] = to_addrs
        server.send_message(msg)
        server.quit()