moviewyrm/bookwyrm/emailing.py

""" send emails """
from django.core.mail import EmailMultiAlternatives
from django.template.loader import get_template

from bookwyrm import models, settings
from bookwyrm.tasks import app
from bookwyrm.settings import DOMAIN


def email_data():
    """fields every email needs"""
    site = models.SiteSettings.objects.get()
    if site.logo_small:
        logo_path = f"/images/{site.logo_small.url}"
    else:
        logo_path = "/static/images/logo-small.png"

    return {
        "site_name": site.name,
        "logo": logo_path,
        "domain": DOMAIN,
        "user": None,
    }


def email_confirmation_email(user):
    """newly registered users confirm email address"""
    data = email_data()
    data["confirmation_code"] = user.confirmation_code
    data["confirmation_link"] = user.confirmation_link
    send_email.delay(user.email, *format_email("confirm", data))


def invite_email(invite_request):
    """send out an invite code"""
    data = email_data()
    data["invite_link"] = invite_request.invite.link
    send_email.delay(invite_request.email, *format_email("invite", data))


def password_reset_email(reset_code):
    """generate a password reset email"""
    data = email_data()
    data["reset_link"] = reset_code.link
    data["user"] = reset_code.user.display_name
    send_email.delay(reset_code.user.email, *format_email("password_reset", data))


def format_email(email_name, data):
    """render the email templates"""
    subject = get_template(f"email/{email_name}/subject.html").render(data).strip()
    html_content = (
        get_template(f"email/{email_name}/html_content.html").render(data).strip()
    )
    text_content = (
        get_template(f"email/{email_name}/text_content.html").render(data).strip()
    )
    return (subject, html_content, text_content)


@app.task(queue="high_priority")
def send_email(recipient, subject, html_content, text_content):
    """use a task to send the email"""
    email = EmailMultiAlternatives(
        subject, text_content, settings.DEFAULT_FROM_EMAIL, [recipient]
    )
    email.attach_alternative(html_content, "text/html")
    email.send()