Codex
157 lines
5.1 KiB
Python
157 lines
5.1 KiB
Python
#!/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
|