mirror of https://github.com/logos-co/open-law.git
144 lines
4.3 KiB
Python
144 lines
4.3 KiB
Python
import os
|
|
import warnings
|
|
|
|
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["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
|