bookwyrm-mastodon/bookwyrm/emailing.py

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

from bookwyrm import models
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 = '/images/{}'.format(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 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, "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.diplay_name
    send_email.delay(reset_code.user.email, "password_reset", data)


@app.task
def send_email(recipient, message_name, data):
    """ use a task to send the email """
    subject = (
        get_template("email/{}/subject.html".format(message_name)).render(data).strip()
    )
    html_content = (
        get_template("email/{}/html_content.html".format(message_name))
        .render(data)
        .strip()
    )
    text_content = (
        get_template("email/{}/text_content.html".format(message_name))
        .render(data)
        .strip()
    )

    email = EmailMultiAlternatives(subject, text_content, None, [recipient])
    email.attach_alternative(html_content, "text/html")
    email.send()