第一次提交

app/api/decorators.py

@@ -0,0 +1,64 @@
+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