from typing import Any from app.api.v1.module_platform.tenant.service import TenantService from app.core.base_schema import AuthSchema, BatchSetAvailable from app.core.exceptions import CustomException from app.utils.excel_util import ExcelUtil from .crud import RoleCRUD from .schema import ( RoleCreateSchema, RoleOutSchema, RolePermissionSettingSchema, RoleQueryParam, RoleUpdateSchema, ) class RoleService: """ 角色管理服务 提供角色 CRUD、权限配置、数据权限范围设置、批量启/禁用、Excel 导出等业务能力。 """ def __init__(self, auth: AuthSchema) -> None: self.auth = auth async def detail(self, id: int) -> RoleOutSchema: """ 获取角色详情 参数: - id (int): 角色ID 返回: - RoleOutSchema: 角色详情响应模型 """ return await RoleCRUD(self.auth).get_or_404(id=id, out_schema=RoleOutSchema) async def get_list( self, search: RoleQueryParam | None = None, order_by: list[dict[str, str]] | None = None, ) -> list[RoleOutSchema]: """ 获取角色列表 参数: - search (RoleQueryParam | None): 查询参数模型 - order_by (list[dict[str, str]] | None): 排序参数列表 返回: - list[RoleOutSchema]: 角色响应模型列表 """ role_list = await RoleCRUD(self.auth).get_list(search=vars(search) if search else None, order_by=order_by) return [RoleOutSchema.model_validate(role) for role in role_list] async def page( self, page_no: int, page_size: int, search: RoleQueryParam | None = None, order_by: list[dict[str, str]] | None = None, ) -> dict: """ 分页查询角色（数据库 OFFSET/LIMIT）。 参数: - page_no (int): 页码（从 1 开始） - page_size (int): 每页条数 - search (RoleQueryParam | None): 查询条件 - order_by (list[dict[str, str]] | None): 排序字段列表 返回: - dict: 分页结果（结构由 ``CRUD.page`` 返回约定） """ offset = (page_no - 1) * page_size return await RoleCRUD(self.auth).page( offset=offset, limit=page_size, order_by=order_by or [{"id": "asc"}], search=vars(search) if search else None, out_schema=RoleOutSchema, ) async def create(self, data: RoleCreateSchema) -> RoleOutSchema: """ 创建角色 参数: - data (RoleCreateSchema): 创建角色模型 返回: - RoleOutSchema: 新创建的角色响应模型 """ role = await RoleCRUD(self.auth).get(name=data.name) if role: raise CustomException(msg="创建失败，该数据已存在") obj = await RoleCRUD(self.auth).get(code=data.code) if obj: raise CustomException(msg="创建失败，编码已存在") # 检查租户配额 await TenantService(self.auth).check_quota(self.auth.tenant_id, "role") new_role = await RoleCRUD(self.auth).create(data=data) return RoleOutSchema.model_validate(new_role) async def update(self, id: int, data: RoleUpdateSchema) -> RoleOutSchema: """ 更新角色 参数: - id (int): 角色ID - data (RoleUpdateSchema): 更新角色模型 返回: - RoleOutSchema: 更新后的角色响应模型 """ _ = await RoleCRUD(self.auth).get_or_404(id=id, msg="更新失败，该数据不存在") exist_role = await RoleCRUD(self.auth).get(name=data.name) if exist_role and exist_role.id != id: raise CustomException(msg="更新失败，名称已存在") exist_code = await RoleCRUD(self.auth).get(code=data.code) if exist_code and exist_code.id != id: raise CustomException(msg="更新失败，角色编码已存在") updated_role = await RoleCRUD(self.auth).update(id=id, data=data) return RoleOutSchema.model_validate(updated_role) async def delete(self, ids: list[int]) -> None: """ 删除角色 参数: - ids (list[int]): 角色ID列表 返回: - None """ if len(ids) < 1: raise CustomException(msg="删除失败，删除对象不能为空") # 批量校验角色存在性 roles = await RoleCRUD(self.auth).get_list(search={"id": ("in", ids)}) if len(roles) != len(ids): raise CustomException(msg="删除失败，部分ID不存在") await RoleCRUD(self.auth).delete(ids=ids) async def set_permission(self, data: RolePermissionSettingSchema) -> None: """ 设置角色权限 参数: - data (RolePermissionSettingSchema): 角色权限设置模型 返回: - None """ # 设置角色菜单权限 await RoleCRUD(self.auth).set_role_menus_crud(role_ids=data.role_ids, menu_ids=data.menu_ids) # 设置数据权限范围 await RoleCRUD(self.auth).set(ids=data.role_ids, data_scope=data.data_scope) # 设置自定义数据权限部门 if data.data_scope == 5 and data.dept_ids: await RoleCRUD(self.auth).set_role_depts_crud(role_ids=data.role_ids, dept_ids=data.dept_ids) else: await RoleCRUD(self.auth).set_role_depts_crud(role_ids=data.role_ids, dept_ids=[]) async def set_available(self, data: BatchSetAvailable) -> None: """ 设置角色可用状态 参数: - data (BatchSetAvailable): 批量设置可用状态模型 返回: - None """ roles = await RoleCRUD(self.auth).get_list(search={"id": ("in", data.ids)}) role_map = {r.id: r for r in roles} for rid in data.ids: if rid not in role_map: raise CustomException(msg="该数据不存在") await RoleCRUD(self.auth).set(ids=data.ids, status=data.status) @staticmethod def export_list(role_list: list[dict[str, Any]]) -> bytes: """ 导出角色列表 参数: - role_list (list[dict[str, Any]]): 角色详情字典列表 返回: - bytes: Excel文件字节流 """ # 字段映射配置 mapping_dict = { "id": "角色编号", "name": "角色名称", "order": "显示顺序", "data_scope": "数据权限", "status": "状态", "description": "备注", "created_time": "创建时间", "updated_time": "更新时间", "created_id": "创建者ID", "updated_id": "更新者ID", } # 数据权限映射 data_scope_map = { 1: "仅本人数据权限", 2: "本部门数据权限", 3: "本部门及以下数据权限", 4: "全部数据权限", 5: "自定义数据权限", } # 处理数据 data = role_list.copy() for item in data: item["status"] = "启用" if item.get("status") == 0 else "停用" item["data_scope"] = data_scope_map.get(item.get("data_scope", 1), "") return ExcelUtil.export_list2excel(list_data=data, mapping_dict=mapping_dict)