forked from loctv/flask-server-starter
/
app.py
61 lines (48 loc) · 1.53 KB
/
app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
# -*- coding: utf-8 -*-
from flask import Flask, render_template
from settings import ProdConfig
from flask.ext.security import Security, SQLAlchemyUserDatastore
from user.models import User, Role
from admin.views import UserView, RoleView
from user.forms import ExtendedRegisterForm
from extensions import (
cache,
admin,
db,
mail,
debug_toolbar,
)
from public.views import bp_public
from user.views import bp_user
def create_app(config_object=ProdConfig):
app = Flask(__name__)
app.config.from_object(config_object)
register_extensions(app)
register_blueprints(app)
register_errorhandlers(app)
return app
def register_extensions(app):
cache.init_app(app)
db.init_app(app)
admin.init_app(app)
register_admin_views(admin)
user_datastore = SQLAlchemyUserDatastore(db, User, Role)
security = Security(app, user_datastore, confirm_register_form=ExtendedRegisterForm)
mail.init_app(app)
debug_toolbar.init_app(app)
return None
def register_blueprints(app):
app.register_blueprint(bp_public)
app.register_blueprint(bp_user)
return None
def register_admin_views(admin):
admin.add_view(UserView(User, db.session))
admin.add_view(RoleView(Role, db.session))
return None
def register_errorhandlers(app):
def render_error(error):
error_code = getattr(error, 'code', 500)
return render_template("{0}.html".format(error_code)), error_code
for errcode in [401, 404, 500]:
app.errorhandler(errcode)(render_error)
return None