Build lightweight AI agent admin

backend-fastapi/app/data_scope_utils.py

@@ -0,0 +1,156 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+数据权限工具函数
+
+提供独立的数据权限过滤功能，供不继承 BaseService 的模块使用
+"""
+from typing import Any, Dict, List, Optional
+
+from sqlalchemy import and_, or_
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from utils.context import get_current_user_info_from_context
+
+
+async def get_data_scope_filter(
+    db: AsyncSession,
+    resource_type: str,
+    user_info: Optional[Dict[str, Any]] = None
+) -> Dict[str, Any]:
+    """
+    获取数据权限过滤条件
+    
+    :param db: 数据库会话
+    :param resource_type: 资源类型（如 "page", "data_source" 等）
+    :param user_info: 用户信息字典，如果不传则自动从上下文获取
+    :return: 数据权限过滤条件字典
+    """
+    # 如果没有传入用户信息，从上下文获取
+    if user_info is None:
+        user_info = get_current_user_info_from_context()
+    
+    # 如果没有用户信息，返回全部数据
+    if not user_info:
+        return {
+            'filter_type': 'all',
+            'scope': 0,
+            'user_id': None,
+            'dept_id': None,
+            'dept_ids': None
+        }
+    
+    # 超级管理员：全部数据
+    if user_info.get("is_superuser", False):
+        return {
+            'filter_type': 'all',
+            'scope': 0,
+            'user_id': None,
+            'dept_id': None,
+            'dept_ids': None
+        }
+    
+    # 如果没有定义资源类型，默认全部数据
+    if not resource_type:
+        return {
+            'filter_type': 'all',
+            'scope': 0,
+            'user_id': None,
+            'dept_id': None,
+            'dept_ids': None
+        }
+    
+    # 使用资源类型绑定的数据权限
+    from core.resource_scope.scope_permission.service import ResourceDataScopeConfigService
+    
+    # 查询资源数据权限配置（支持多角色）
+    config_dict = await ResourceDataScopeConfigService.get_resource_data_scope(
+        db=db,
+        role_ids=user_info.get("role_ids", []),
+        resource_type=resource_type,
+        is_superuser=user_info.get("is_superuser", False)
+    )
+    
+    # 填充用户和部门信息
+    if config_dict['filter_type'] == 'self':
+        config_dict['user_id'] = user_info.get("user_id")
+    elif config_dict['filter_type'] == 'dept':
+        config_dict['dept_id'] = user_info.get("dept_id")
+    elif config_dict['filter_type'] == 'dept_and_children':
+        # 获取部门树
+        if user_info.get("dept_id"):
+            from core.auth.dept.service import DeptService
+            dept_ids = await DeptService.get_dept_and_children_ids(db, user_info.get("dept_id"))
+            config_dict['dept_ids'] = dept_ids
+        else:
+            config_dict['dept_ids'] = []
+    
+    return config_dict
+
+
+def apply_data_scope_to_conditions(
+    model,
+    data_scope_filter: Dict[str, Any],
+    dept_field: str = "sys_dept_id",
+    user_field: str = "sys_creator_id"
+) -> List[Any]:
+    """
+    根据数据权限过滤条件生成 SQLAlchemy 条件列表
+    
+    注意：当 sys_creator_id 或 sys_dept_id 为空时，记录对所有人可见
+    
+    :param model: SQLAlchemy 模型类
+    :param data_scope_filter: 数据权限过滤条件
+    :param dept_field: 部门字段名
+    :param user_field: 用户字段名
+    :return: SQLAlchemy 条件列表
+    """
+    conditions = []
+    filter_type = data_scope_filter.get('filter_type')
+    
+    # 全部数据：不添加过滤条件
+    if filter_type == 'all':
+        return conditions
+    
+    # 仅本人数据：匹配创建人 OR 创建人为空（对所有人可见）
+    if filter_type == 'self':
+        if hasattr(model, user_field):
+            user_field_obj = getattr(model, user_field)
+            conditions.append(
+                or_(
+                    user_field_obj == data_scope_filter['user_id'],
+                    user_field_obj.is_(None)  # 创建人为空时所有人可见
+                )
+            )
+        return conditions
+    
+    # 本部门数据：匹配部门 OR 部门为空（对所有人可见）
+    if filter_type == 'dept':
+        if hasattr(model, dept_field):
+            dept_field_obj = getattr(model, dept_field)
+            conditions.append(
+                or_(
+                    dept_field_obj == data_scope_filter['dept_id'],
+                    dept_field_obj.is_(None)  # 部门为空时所有人可见
+                )
+            )
+        return conditions
+    
+    # 本部门及下级部门数据 / 自定义数据
+    if filter_type in ('dept_and_children', 'custom'):
+        if hasattr(model, dept_field):
+            dept_field_obj = getattr(model, dept_field)
+            dept_ids = data_scope_filter.get('dept_ids', [])
+            if dept_ids:
+                conditions.append(
+                    or_(
+                        dept_field_obj.in_(dept_ids),
+                        dept_field_obj.is_(None)  # 部门为空时所有人可见
+                    )
+                )
+            else:
+                # 如果没有部门ID，只返回部门为空的记录
+                conditions.append(dept_field_obj.is_(None))
+        return conditions
+    
+    return conditions