certidude/certidude/mailer.py

import click
import os
import smtplib
from certidude.user import User
from markdown import markdown
from jinja2 import Environment, PackageLoader
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText
from email.mime.base import MIMEBase
from email.header import Header
from urllib.parse import urlparse

env = Environment(loader=PackageLoader("certidude", "templates/mail"))

def send(template, to=None, secondary=None, include_admins=True, attachments=(), **context):
    from certidude import authority, config

    recipients = ()
    if include_admins:
        recipients = tuple(User.objects.filter_admins())
    if to:
        recipients = (to,) + recipients
    if secondary:
        recipients = (secondary,) + recipients


    click.echo("Sending e-mail %s to %s" % (template, recipients))

    subject, text = env.get_template(template).render(context).split("\n\n", 1)
    html = markdown(text)

    msg = MIMEMultipart("alternative")
    msg["Subject"] = Header(subject)
    msg["From"] = Header(config.MAILER_NAME)
    msg["From"].append("<%s>" % config.MAILER_ADDRESS)

    msg["To"] = Header()
    for user in recipients:
        if isinstance(user, User):
            full_name, user = user.format()
            if full_name:
                msg["To"].append(full_name)
        msg["To"].append(user)
        msg["To"].append(", ")

    part1 = MIMEText(text, "plain", "utf-8")
    part2 = MIMEText(html, "html", "utf-8")

    msg.attach(part1)
    msg.attach(part2)

    for attachment, content_type, suggested_filename in attachments:
        part = MIMEBase(*content_type.split("/"))
        part.add_header('Content-Disposition', 'attachment', filename=suggested_filename)
        part.set_payload(attachment)
        msg.attach(part)

    if config.MAILER_ADDRESS:
        click.echo("Sending to: %s" % msg["to"])
        conn = smtplib.SMTP("localhost")
        conn.sendmail(config.MAILER_ADDRESS, [u.mail if isinstance(u, User) else u for u in recipients], msg.as_string())