flaskbb/flaskbb/email.py

# -*- coding: utf-8 -*-
"""
    flaskbb.email
    ~~~~~~~~~~~~~

    This module adds the functionality to send emails

    :copyright: (c) 2014 by the FlaskBB Team.
    :license: BSD, see LICENSE for more details.
"""
from flask import render_template
from flask_mail import Message
from flask_babelplus import lazy_gettext as _

from flaskbb.extensions import mail, celery
from flaskbb.utils.tokens import make_token


@celery.task
def send_reset_token(user):
    """Sends the reset token to the user's email address.

    :param user: The user object to whom the email should be sent.
    """
    token = make_token(user=user, operation="reset_password")
    send_email(
        subject=_("Password Recovery Confirmation"),
        recipients=[user.email],
        text_body=render_template(
            "email/reset_password.txt",
            user=user,
            token=token
        ),
        html_body=render_template(
            "email/reset_password.html",
            user=user,
            token=token
        )
    )


@celery.task
def send_activation_token(user):
    """Sends the activation token to the user's email address.

    :param user: The user object to whom the email should be sent.
    """
    token = make_token(user=user, operation="activate_account")
    send_email(
        subject=_("Account Activation"),
        recipients=[user.email],
        text_body=render_template(
            "email/activate_account.txt",
            user=user,
            token=token
        ),
        html_body=render_template(
            "email/activate_account.html",
            user=user,
            token=token
        )
    )


@celery.task
def send_async_email(*args, **kwargs):
    send_email(*args, **kwargs)


def send_email(subject, recipients, text_body, html_body, sender=None):
    """Sends an email to the given recipients.

    :param subject: The subject of the email.
    :param recipients: A list of recipients.
    :param text_body: The text body of the email.
    :param html_body: The html body of the email.
    :param sender: A two-element tuple consisting of name and address.
                   If no sender is given, it will fall back to the one you
                   have configured with ``MAIL_DEFAULT_SENDER``.
    """
    msg = Message(subject, recipients=recipients, sender=sender)
    msg.body = text_body
    msg.html = html_body
    mail.send(msg)