64 lines
2.4 KiB
Python
64 lines
2.4 KiB
Python
|
from flask import jsonify
|
|||
|
|
from flask_login import current_user
|
|||
|
|
import functools
|
|||
|
|
|
|||
|
|
|
|||
|
|
def require_login(f):
|
|||
|
|
"""登录用户权限验证装饰器（普通管理员和超级管理员都可以访问）
|
|||
|
|
|
|||
|
|
注意：对于API端点，不使用@login_required装饰器，因为它会重定向到登录页面
|
|||
|
|
而不是返回JSON错误响应。我们直接检查认证状态并返回JSON。
|
|||
|
|
"""
|
|||
|
|
@functools.wraps(f)
|
|||
|
|
def decorated_function(*args, **kwargs):
|
|||
|
|
# 检查用户是否已认证
|
|||
|
|
# Flask-Login 的 current_user 在未登录时是一个 AnonymousUserMixin 实例
|
|||
|
|
# 它的 is_authenticated 属性为 False
|
|||
|
|
if not hasattr(current_user, 'is_authenticated') or not current_user.is_authenticated:
|
|||
|
|
return jsonify({
|
|||
|
|
'success': False,
|
|||
|
|
'message': '需要登录'
|
|||
|
|
}), 401
|
|||
|
|
|
|||
|
|
# 检查账号是否激活（is_active 是 Admin 模型的属性）
|
|||
|
|
if hasattr(current_user, 'is_active') and not current_user.is_active:
|
|||
|
|
return jsonify({
|
|||
|
|
'success': False,
|
|||
|
|
'message': '账号已被禁用'
|
|||
|
|
}), 403
|
|||
|
|
|
|||
|
|
return f(*args, **kwargs)
|
|||
|
|
return decorated_function
|
|||
|
|
|
|||
|
|
|
|||
|
|
def require_admin(f):
|
|||
|
|
"""超级管理员权限验证装饰器（只有超级管理员可以访问）
|
|||
|
|
|
|||
|
|
注意：对于API端点，不使用@login_required装饰器，因为它会重定向到登录页面
|
|||
|
|
而不是返回JSON错误响应。我们直接检查认证状态并返回JSON。
|
|||
|
|
"""
|
|||
|
|
@functools.wraps(f)
|
|||
|
|
def decorated_function(*args, **kwargs):
|
|||
|
|
# 检查用户是否已认证
|
|||
|
|
if not hasattr(current_user, 'is_authenticated') or not current_user.is_authenticated:
|
|||
|
|
return jsonify({
|
|||
|
|
'success': False,
|
|||
|
|
'message': '需要登录'
|
|||
|
|
}), 401
|
|||
|
|
|
|||
|
|
# 检查账号是否激活
|
|||
|
|
if hasattr(current_user, 'is_active') and not current_user.is_active:
|
|||
|
|
return jsonify({
|
|||
|
|
'success': False,
|
|||
|
|
'message': '账号已被禁用'
|
|||
|
|
}), 403
|
|||
|
|
|
|||
|
|
# 检查是否为超级管理员
|
|||
|
|
if not hasattr(current_user, 'is_super_admin') or not current_user.is_super_admin():
|
|||
|
|
return jsonify({
|
|||
|
|
'success': False,
|
|||
|
|
'message': '需要超级管理员权限'
|
|||
|
|
}), 403
|
|||
|
|
|
|||
|
|
return f(*args, **kwargs)
|
|||
|
|
return decorated_function
|