notifications-admin/app/utils/nunjucks_jinja/flask_ext.py

from flask.templating import Environment as FlaskEnvironment
from jinja2 import select_autoescape

from app.utils.nunjucks_jinja.templates import Environment as NunjucksEnvironment
from app.utils.nunjucks_jinja.templates import (
    NunjucksExtension,
    NunjucksUndefined,
)


class Environment(NunjucksEnvironment, FlaskEnvironment):
    pass


def init_nunjucks_environment(app):
    """Use the Nunjucks-compatible Jinja environment in a Flask app

    >>> from flask import Flask
    >>> app = Flask("notifications_service")
    >>> init_nunjucks_environment(app)
    """
    app.jinja_environment = Environment
    app.select_jinja_autoescape = select_autoescape(
        ("html", "htm", "xml", "xhtml", "njk")
    )
    jinja_options = app.jinja_options.copy()
    jinja_options["extensions"].append(NunjucksExtension)
    jinja_options["undefined"] = NunjucksUndefined
    app.jinja_options = jinja_options
    return app