# project/server/__init__.py ################# #### imports #### ################# import os from flask import Flask, render_template from flask_login import LoginManager from flask_bcrypt import Bcrypt from flask_debugtoolbar import DebugToolbarExtension from flask_bootstrap import Bootstrap from flask_sqlalchemy import SQLAlchemy ################ #### config #### ################ app = Flask( __name__, template_folder='../client/templates', static_folder='../client/static' ) app_settings = os.getenv('APP_SETTINGS', 'project.server.config.DevelopmentConfig') app.config.from_object(app_settings) #################### #### extensions #### #################### login_manager = LoginManager() login_manager.init_app(app) bcrypt = Bcrypt(app) toolbar = DebugToolbarExtension(app) bootstrap = Bootstrap(app) db = SQLAlchemy(app) ################### ### blueprints #### ################### from project.server.user.views import user_blueprint from project.server.main.views import main_blueprint app.register_blueprint(user_blueprint) app.register_blueprint(main_blueprint) ################### ### flask-login #### ################### from project.server.models import User login_manager.login_view = "user.login" login_manager.login_message_category = 'danger' @login_manager.user_loader def load_user(user_id): return User.query.filter(User.id == int(user_id)).first() ######################## #### error handlers #### ######################## @app.errorhandler(401) def unauthorized_page(error): return render_template("errors/401.html"), 401 @app.errorhandler(403) def forbidden_page(error): return render_template("errors/403.html"), 403 @app.errorhandler(404) def page_not_found(error): return render_template("errors/404.html"), 404 @app.errorhandler(500) def server_error_page(error): return render_template("errors/500.html"), 500