Build lightweight AI agent admin

2026-06-08 18:14:59 +08:00
commit e164840f43
2530 changed files with 435693 additions and 0 deletions
@@ -0,0 +1,77 @@
+"""
+上下文管理器 - 用于在整个请求生命周期中共享数据
+"""
+from contextvars import ContextVar
+from typing import Optional, Dict, Any
+
+# 当前用户信息上下文
+current_user_context: ContextVar[Optional[Dict[str, Any]]] = ContextVar('current_user', default=None)
+
+
+def get_current_user_id_from_context() -> Optional[str]:
+    """
+    从上下文中获取当前用户ID
+    
+    :return: 用户ID，如果未设置则返回None
+    """
+    user_info = current_user_context.get()
+    return user_info.get('user_id') if user_info else None
+
+
+def get_current_user_info_from_context() -> Optional[Dict[str, Any]]:
+    """
+    从上下文中获取当前用户完整信息
+    
+    :return: 用户信息字典，包含 user_id, role_id, dept_id, is_superuser 等
+    """
+    return current_user_context.get()
+
+
+def set_current_user_context(
+    user_id: str,
+    role_id: Optional[str] = None,
+    dept_id: Optional[str] = None,
+    is_superuser: bool = False,
+    username: Optional[str] = None,
+    request_path: Optional[str] = None,
+    http_method: Optional[str] = None,
+    role_ids: Optional[list] = None
+) -> None:
+    """
+    设置当前用户信息到上下文
+    
+    :param user_id: 用户ID
+    :param role_id: 角色ID（单个，向后兼容）
+    :param dept_id: 部门ID
+    :param is_superuser: 是否超级管理员
+    :param username: 用户名
+    :param request_path: 请求路径
+    :param http_method: HTTP方法
+    :param role_ids: 角色ID列表（多个角色）
+    """
+    # 如果提供了 role_ids，使用它；否则从 role_id 构建
+    if role_ids is None and role_id:
+        role_ids = [role_id]
+    elif role_ids is None:
+        role_ids = []
+    
+    current_user_context.set({
+        'user_id': user_id,
+        'role_id': role_id,  # 保持向后兼容
+        'role_ids': role_ids,  # 新增：支持多角色
+        'dept_id': dept_id,
+        'is_superuser': is_superuser,
+        'username': username,
+        'request_path': request_path,
+        'http_method': http_method
+    })
+
+
+def clear_current_user_context() -> None:
+    """清除当前用户上下文"""
+    current_user_context.set(None)
+
+
+# 保持向后兼容的别名
+set_current_user_id_context = lambda user_id: set_current_user_context(user_id)
+clear_current_user_id_context = clear_current_user_context