管理后台 server 使用蓝图模式,降低维护成本,降低修改功能时对其他模块的影响
This commit is contained in:
39
admin/dashboard/blueprints/auth.py
Normal file
39
admin/dashboard/blueprints/auth.py
Normal file
@@ -0,0 +1,39 @@
|
||||
from flask import Blueprint, render_template, request, redirect, url_for, session, current_app
|
||||
from functools import wraps
|
||||
|
||||
# 创建认证蓝图
|
||||
auth_bp = Blueprint('auth', __name__)
|
||||
|
||||
# 登录检查装饰器
|
||||
def login_required(f):
|
||||
@wraps(f)
|
||||
def decorated_function(*args, **kwargs):
|
||||
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():
|
||||
error = None
|
||||
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'))
|
||||
Reference in New Issue
Block a user