修改数据库初始化逻辑,完善权限,废弃flaskenv
This commit is contained in:
@@ -7,7 +7,7 @@ from .init_error_views import init_error_views
|
||||
from .init_mail import init_mail, mail as flask_mail
|
||||
from .init_apscheduler import init_scheduler
|
||||
from .init_upload import init_upload
|
||||
from .init_dotenv import init_dotenv
|
||||
from .init_migrate import init_migrate
|
||||
|
||||
|
||||
def init_plugs(app: Flask) -> None:
|
||||
@@ -18,4 +18,4 @@ def init_plugs(app: Flask) -> None:
|
||||
init_mail(app)
|
||||
init_scheduler(app)
|
||||
init_upload(app)
|
||||
init_dotenv()
|
||||
init_migrate(app)
|
||||
@@ -1,14 +0,0 @@
|
||||
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)
|
||||
|
||||
def init_dotenv():
|
||||
if os.path.exists(flask_env_path):
|
||||
load_dotenv(flask_env_path)
|
||||
@@ -0,0 +1,11 @@
|
||||
from flask import Flask
|
||||
|
||||
from applications.extensions.init_sqlalchemy import db
|
||||
from flask_migrate import Migrate
|
||||
from applications.models import *
|
||||
|
||||
migrate = Migrate()
|
||||
|
||||
|
||||
def init_migrate(app: Flask):
|
||||
migrate.init_app(app, db)
|
||||
@@ -1,7 +1,11 @@
|
||||
from flask import session
|
||||
from flask import session, current_app
|
||||
from flask_login import current_user
|
||||
|
||||
|
||||
def init_template_directives(app):
|
||||
@app.template_global()
|
||||
def authorize(power):
|
||||
return bool(power in session.get('permissions'))
|
||||
if current_user.username != current_app.config.get("SUPERADMIN"):
|
||||
return bool(power in session.get('permissions'))
|
||||
else:
|
||||
return True
|
||||
|
||||
Reference in New Issue
Block a user