refactor(重构程序&目录结构)

extensions/__init__.py

@@ -0,0 +1,17 @@
+from flask import Flask
+
+from .init_databases import register_script
+from .init_sqlalchemy import db, init_databases
+from .init_login import init_login_manager
+from .init_template_directives import init_template_directives
+from .init_error_views import init_error_views
+
+
+def init_plugs(app: Flask) -> None:
+    init_login_manager(app)
+    init_databases(app)
+    init_template_directives(app)
+    init_error_views(app)
+
+    # 生成测试数据的命令，生成环境可以注释
+    register_script(app)

extensions/init_databases.py

@@ -0,0 +1,82 @@
+from flask import Flask
+import re
+from datetime import datetime
+
+date_str = re.compile('\d\d\d\d-\d\d-\d\d \d\d:\d\d:\d\d$')
+
+
+def add_data(data_list, obj):
+    from extensions import db
+
+    for _data in data_list:
+        dept = obj()
+        for key, value in _data.items():
+
+            if isinstance(value, str) and date_str.match(value):
+                value = datetime.strptime(value, "%Y-%m-%d %H:%M:%S")
+
+            setattr(dept, key, value)
+        db.session.add(dept)
+    db.session.commit()
+
+
+def register_script(app: Flask):
+    @app.cli.command()
+    def init_db():
+        import json
+        import os
+
+        path = os.path.dirname(os.path.abspath(__name__))
+        path = os.path.join(path, 'static', 'data')
+        cp_dept_data_list = json.loads(open(os.path.join(path, 'cp_dept.json'), encoding='utf-8').read())
+        cp_user_data_list = json.loads(open(os.path.join(path, 'cp_user.json'), encoding='utf-8').read())
+        file_photo_data_list = json.loads(open(os.path.join(path, 'file_photo.json'), encoding='utf-8').read())
+        rt_power_data_list = json.loads(open(os.path.join(path, 'rt_power.json'), encoding='utf-8').read())
+        rt_role_data_list = json.loads(open(os.path.join(path, 'rt_role.json'), encoding='utf-8').read())
+        rt_role_power_data_list = json.loads(open(os.path.join(path, 'rt_role_power.json'), encoding='utf-8').read())
+        rt_user_role_data_list = json.loads(open(os.path.join(path, 'rt_user_role.json'), encoding='utf-8').read())
+
+        """数据库初始化"""
+
+        # 创建化部门数据
+        from models import DepartmentModels
+        add_data(cp_dept_data_list, DepartmentModels)
+
+        # 图片数据
+        from models import PhotoModels
+
+        add_data(file_photo_data_list, PhotoModels)
+
+        # 初始化权限表数据
+        from models import RightModels
+
+        add_data(rt_power_data_list, RightModels)
+        # 初始化角色表
+        from models import RoleModels
+
+        # 角色权限关系表
+        from extensions import db
+        for data in rt_role_power_data_list:
+            db.session.execute('insert into rt_role_power VALUES (%s, %s, %s);' % tuple(data))
+        db.session.commit()
+
+        add_data(rt_role_data_list, RoleModels)
+
+        # 管理员用户
+        from models import UserModels
+
+        add_data(cp_user_data_list, UserModels)
+
+        # 用户角色表
+        from extensions import db
+
+        for data in rt_user_role_data_list:
+            db.session.execute('insert into rt_user_role VALUES (%s, %s, %s);' % tuple(data))
+        db.session.commit()
+
+    @app.cli.command()
+    def turn():
+        """清空数据库"""
+        from extensions import db
+        db.drop_all()
+        db.create_all()

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)

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

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 = 'index.login'
+    login_manager.login_message = u'请登录以访问此页面'
+
+    @login_manager.user_loader
+    def load_user(user_id):
+        from models import UserModels
+        user = UserModels.query.get(int(user_id))
+        return user

extensions/init_sqlalchemy.py

@@ -0,0 +1,11 @@
+from flask import Flask
+from flask_sqlalchemy import SQLAlchemy
+from flask_migrate import Migrate
+
+db = SQLAlchemy()
+migrate = Migrate()
+
+
+def init_databases(app: Flask):
+    db.init_app(app)
+    migrate.init_app(app, db)

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'))