Source code for flask_wings.FlaskWings

""" Boilerplate for a Flask extension class. """

import os

import jinja2
from .views import views


class FlaskWings:
    """ Boilerplate Flask extension that overrides templates and views."""

    def __init__(self, app):
        """ Add the default templates directory and default views
        to the parent application.
        """
        templates_dir = os.path.join(
            os.path.dirname(os.path.realpath(__file__)), 'templates')
        my_loader = jinja2.ChoiceLoader([
            app.jinja_loader,
            jinja2.FileSystemLoader(templates_dir),
        ])
        app.jinja_loader = my_loader
        app.register_blueprint(views)