open-law/app/__init__.py

import os
import warnings
from uuid import uuid4

from flask import Flask, render_template
from flask_sqlalchemy import SQLAlchemy
from flask_login import LoginManager
from werkzeug.exceptions import HTTPException
from flask_migrate import Migrate
from flask_admin import Admin

from app.logger import log

# instantiate extensions
login_manager = LoginManager()
db = SQLAlchemy()
migration = Migrate()


def create_app(environment="development"):
    from config import config
    from app.views import (
        main_blueprint,
        auth_blueprint,
        user_blueprint,
        book_blueprint,
        home_blueprint,
        vote_blueprint,
        approve_blueprint,
        star_blueprint,
        permissions_blueprint,
        search_blueprint,
        notifications_blueprint,
    )
    from app import models as m

    # Instantiate app.
    app = Flask(__name__)

    # Set app config.
    env = os.environ.get("APP_ENV", environment)
    configuration = config(env)
    app.config.from_object(configuration)
    configuration.configure(app)
    log(log.INFO, "Configuration: [%s]", configuration.ENV)

    # Set up extensions.
    db.init_app(app)
    migration.init_app(app, db)
    login_manager.init_app(app)

    # Register blueprints.
    app.register_blueprint(auth_blueprint)
    app.register_blueprint(main_blueprint)
    app.register_blueprint(user_blueprint)
    app.register_blueprint(book_blueprint)
    app.register_blueprint(home_blueprint)
    app.register_blueprint(vote_blueprint)
    app.register_blueprint(approve_blueprint)
    app.register_blueprint(star_blueprint)
    app.register_blueprint(permissions_blueprint)
    app.register_blueprint(search_blueprint)
    app.register_blueprint(notifications_blueprint)

    # Set up flask login.
    @login_manager.user_loader
    def get_user(id):
        return m.User.query.get(int(id))

    login_manager.login_view = "auth.login"
    login_manager.login_message_category = "info"
    login_manager.anonymous_user = m.AnonymousUser

    # Jinja globals
    from app.controllers.jinja_globals import (
        form_hidden_tag,
        display_inline_elements,
        build_qa_url_using_interpretation,
        recursive_render,
        has_permission,
    )

    app.jinja_env.globals["type"] = type
    app.jinja_env.globals["m"] = m
    app.jinja_env.globals["str"] = str
    app.jinja_env.globals["uuid"] = uuid4

    app.jinja_env.globals["Access"] = m.Permission.Access
    app.jinja_env.globals["EntityType"] = m.Permission.Entity

    app.jinja_env.globals["form_hidden_tag"] = form_hidden_tag
    app.jinja_env.globals["display_inline_elements"] = display_inline_elements
    app.jinja_env.globals["build_qa_url"] = build_qa_url_using_interpretation
    app.jinja_env.globals["recursive_render"] = recursive_render
    app.jinja_env.globals["has_permission"] = has_permission

    # Error handlers.
    @app.errorhandler(HTTPException)
    def handle_http_error(exc):
        return render_template("error.html", error=exc), exc.code

    # flask admin
    from app.controllers.admin import (
        CustomAdminIndexView,
        UsersView,
        BooksView,
        CollectionsView,
        SectionsView,
        InterpretationView,
        CommentView,
        TagView,
        BookContributorView,
    )

    admin = Admin(
        app,
        name="Open Law Admin",
        template_mode="bootstrap3",
        index_view=CustomAdminIndexView(),
    )

    with warnings.catch_warnings():
        warnings.filterwarnings("ignore", "Fields missing from ruleset", UserWarning)

        for view in [
            UsersView(m.User, db.session, name="User", endpoint="/user_"),
            BooksView(m.Book, db.session, name="Book", endpoint="/book_"),
            CollectionsView(
                m.Collection, db.session, name="Collection", endpoint="/collection_"
            ),
            SectionsView(m.Section, db.session, name="Section", endpoint="/section_"),
            InterpretationView(
                m.Interpretation,
                db.session,
                name="Interpretation",
                endpoint="/interpretation_",
            ),
            CommentView(m.Comment, db.session, name="Comment", endpoint="/comment_"),
            TagView(m.Tag, db.session, name="Tag", endpoint="/tag_"),
            BookContributorView(
                m.BookContributor,
                db.session,
                name="BookContributor",
                endpoint="/book_contributor_",
            ),
        ]:
            admin.add_view(view)

    return app