格式化代码
This commit is contained in:
@@ -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'))
|
||||
|
||||
Reference in New Issue
Block a user