Source code for invenio_mail.api
# -*- coding: utf-8 -*-
#
# This file is part of Invenio.
# Copyright (C) 2016-2018 CERN.
#
# Invenio is free software; you can redistribute it and/or modify it
# under the terms of the MIT License; see LICENSE file for more details.
"""Template based messages."""
from __future__ import absolute_import, print_function
from flask import render_template
from flask_mail import Message
[docs]class TemplatedMessage(Message):
"""Siplify creation of templated messages."""
def __init__(self, template_body=None, template_html=None, ctx=None,
**kwargs):
r"""Build message body and HTML based on provided templates.
Provided templates can use keyword arguments ``body`` and ``html``
respectively.
:param template_body: Path to the text template.
:param template_html: Path to the html template.
:param ctx: A mapping containing additional information passed to the
template.
:param \*\*kwargs: Keyword arguments as defined in
:class:`flask_mail.Message`.
"""
ctx = ctx if ctx else {}
if template_body:
kwargs['body'] = render_template(
template_body, body=kwargs.get('body'), **ctx
)
if template_html:
kwargs['html'] = render_template(
template_html, html=kwargs.get('html'), **ctx
)
super(TemplatedMessage, self).__init__(**kwargs)