docker-k8s-devops/chapter5/labs/flask-skeleton/skeleton/server/__init__.py

64 lines
1.5 KiB
Python
Executable File

# Copyright 2016 Cisco Systems, Inc.
# All rights reserved.
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
app = Flask(
__name__,
template_folder='../client/templates',
static_folder='../client/static'
)
app_settings = os.getenv('APP_SETTINGS', 'skeleton.server.config.DevelopmentConfig')
app.config.from_object(app_settings)
login_manager = LoginManager()
login_manager.init_app(app)
bcrypt = Bcrypt(app)
toolbar = DebugToolbarExtension(app)
bootstrap = Bootstrap(app)
db = SQLAlchemy(app)
from skeleton.server.user.views import user_blueprint
from skeleton.server.main.views import main_blueprint
app.register_blueprint(user_blueprint)
app.register_blueprint(main_blueprint)
from skeleton.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()
@app.errorhandler(401)
def forbidden_page_401(error):
return render_template("errors/401.html"), 401
@app.errorhandler(403)
def forbidden_page_403(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