covot/app/notifier.py

from email.message import EmailMessage
from secrets import token_hex
from smtplib import SMTP_SSL

from constants import DOMAIN, PASSWORD, USERNAME
from database.crud import save_attribute


def initialize_smtp(domain, username, password):
    server = SMTP_SSL(domain)
    server.login(username, password)
    return server


def format_message(sender, recipient):
    code = token_hex(4)
    body = "La clave es {}. Envíele esta clave al bot.".format()
    message = EmailMessage()
    message["From"] = sender
    message["To"] = recipient
    message["Subject"] = "Clave para verificar su identidad"
    message.set_content(body)
    return message, code


def send_mail(recipient):
    server = initialize_smtp(domain=DOMAIN, username=USERNAME, password=PASSWORD)
    message, code = format_message(sender=USERNAME, recipient=recipient)
    save_attribute(attribute="codigo", data=code)
    server.send_message(msg=message)
    server.close()