格式化代码

This commit is contained in:
liuwei
2025-06-04 17:52:41 +08:00
parent cd2f8b9679
commit 94e0c16e7b
9 changed files with 148 additions and 110 deletions
+8 -4
View File
@@ -4,6 +4,7 @@ from functools import wraps
# 创建认证蓝图
auth_bp = Blueprint('auth', __name__)
# 登录检查装饰器
def login_required(f):
@wraps(f)
@@ -11,8 +12,10 @@ def login_required(f):
if not session.get('logged_in'):
return redirect(url_for('auth.login'))
return f(*args, **kwargs)
return decorated_function
# 登录页面
@auth_bp.route('/login', methods=['GET', 'POST'])
def login():
@@ -20,20 +23,21 @@ def login():
if request.method == 'POST':
username = request.form['username']
password = request.form['password']
# 从应用上下文获取服务器实例,而不是从蓝图对象
server = current_app.dashboard_server
if username == server.username and password == server.password:
session['logged_in'] = True
return redirect(url_for('main.index'))
else:
error = '用户名或密码错误'
return render_template('login.html', error=error)
# 登出
@auth_bp.route('/logout')
def logout():
session.pop('logged_in', None)
return redirect(url_for('auth.login'))
return redirect(url_for('auth.login'))