调整目录结构

applications/extensions/__init__.py

@@ -0,0 +1,15 @@
+from flask import Flask
+
+from .init_sqlalchemy import db, ma, init_databases
+from .init_login import init_login_manager
+from .init_debug_tool import init_debug_tool
+from .init_template_directives import init_template_directives
+from .init_error_views import init_error_views
+
+
+def init_plugs(app: Flask) -> None:
+    init_debug_tool(app)
+    init_login_manager(app)
+    init_databases(app)
+    init_template_directives(app)
+    init_error_views(app)

applications/extensions/init_debug_tool.py

@@ -0,0 +1,7 @@
+from flask import Flask
+from flask_debugtoolbar import DebugToolbarExtension
+
+
+def init_debug_tool(app: Flask):
+    toolbar = DebugToolbarExtension()
+    toolbar.init_app(app)

applications/extensions/init_dotenv.py

@@ -0,0 +1,12 @@
+import os
+from dotenv import load_dotenv
+
+root_path = os.path.abspath(os.path.dirname(__file__)).split('applications')[0]
+dot_env_path = os.path.join(root_path, '.env')
+flask_env_path = os.path.join(root_path, '.flaskenv')
+
+if os.path.exists(dot_env_path):
+    load_dotenv(dot_env_path)
+
+if os.path.exists(flask_env_path):
+    load_dotenv(flask_env_path)

applications/extensions/init_error_views.py

@@ -0,0 +1,15 @@
+from flask import render_template
+
+
+def init_error_views(app):
+    @app.errorhandler(403)
+    def page_not_found(e):
+        return render_template('errors/403.html'), 403
+
+    @app.errorhandler(404)
+    def page_not_found(e):
+        return render_template('errors/404.html'), 404
+
+    @app.errorhandler(500)
+    def internal_server_error(e):
+        return render_template('errors/500.html'), 500

applications/extensions/init_login.py

@@ -0,0 +1,15 @@
+from flask_login import LoginManager
+
+
+def init_login_manager(app):
+    login_manager = LoginManager()
+    login_manager.init_app(app)
+
+    login_manager.login_view = 'adminIndex.login'
+    login_manager.login_message = u'请登录以访问此页面'
+
+    @login_manager.user_loader
+    def load_user(user_id):
+        from applications.models import User
+        user = User.query.get(int(user_id))
+        return user

applications/extensions/init_sqlalchemy.py

@@ -0,0 +1,54 @@
+from flask import Flask
+from flask_sqlalchemy import SQLAlchemy
+from flask_marshmallow import Marshmallow
+from marshmallow import fields
+from marshmallow.validate import (
+    URL, Email, Range, Length, Equal, Regexp,
+    Predicate, NoneOf, OneOf, ContainsOnly
+)
+
+URL.default_message = '无效的链接'
+Email.default_message = '无效的邮箱地址'
+Range.message_min = '不能小于{min}'
+Range.message_max = '不能小于{max}'
+Range.message_all = '不能超过{min}和{max}这个范围'
+Length.message_min = '长度不得小于{min}位'
+Length.message_max = '长度不得大于{max}位'
+Length.message_all = '长度不能超过{min}和{max}这个范围'
+Length.message_equal = '长度必须等于{equal}位'
+Equal.default_message = '必须等于{other}'
+Regexp.default_message = '非法输入'
+Predicate.default_message = '非法输入'
+NoneOf.default_message = '非法输入'
+OneOf.default_message = '无效的选择'
+ContainsOnly.default_message = '一个或多个无效的选择'
+
+fields.Field.default_error_messages = {
+    "required": "缺少必要数据",
+    "null": "数据不能为空",
+    "validator_failed": "非法数据",
+}
+
+fields.Str.default_error_messages = {
+    'invalid': "不是合法文本"
+}
+
+fields.Int.default_error_messages = {
+    "invalid": "不是合法整数"
+}
+
+fields.Number.default_error_messages = {
+    "invalid": "不是合法数字"
+}
+
+fields.Boolean.default_error_messages = {
+    "invalid": "不是合法布尔值"
+}
+
+db = SQLAlchemy()
+ma = Marshmallow()
+
+
+def init_databases(app: Flask):
+    db.init_app(app)
+    ma.init_app(app)

applications/extensions/init_template_directives.py

@@ -0,0 +1,7 @@
+from flask import session
+
+
+def init_template_directives(app):
+    @app.template_global()
+    def authorize(power):
+        return bool(power in session.get('permissions'))