104 lines
3.4 KiB
Python
104 lines
3.4 KiB
Python
from pydantic import (
|
|
BaseModel,
|
|
ConfigDict,
|
|
Field,
|
|
field_validator,
|
|
model_validator,
|
|
)
|
|
|
|
from app.api.v1.module_system.dept.schema import DeptOutSchema
|
|
from app.api.v1.module_system.menu.schema import MenuOutSchema
|
|
from app.core.base_schema import BaseQueryParam, BaseSchema, UserByQueryParam, UserBySchema
|
|
from app.core.validator import (
|
|
role_permission_request_validator,
|
|
validate_required_code,
|
|
)
|
|
|
|
|
|
class RoleCreateSchema(BaseModel):
|
|
"""角色创建模型
|
|
"""
|
|
|
|
name: str = Field(..., min_length=1, max_length=64, description="角色名称")
|
|
code: str = Field(..., min_length=2, max_length=64, description="角色编码")
|
|
order: int | None = Field(default=1, ge=0, description="显示排序")
|
|
data_scope: int | None = Field(
|
|
default=1,
|
|
ge=1,
|
|
le=3,
|
|
description="数据权限范围(1:仅本人 2:本部门及以下 3:全部)",
|
|
)
|
|
status: int = Field(default=0, ge=0, le=1, description="状态(0:启动 1:停用)")
|
|
description: str | None = Field(default=None, max_length=255, description="描述")
|
|
|
|
@field_validator("code")
|
|
@classmethod
|
|
def validate_code(cls, value: str):
|
|
"""校验角色编码：字母开头，2-64 位，仅含字母/数字/下划线"""
|
|
return validate_required_code(value)
|
|
|
|
@field_validator("status")
|
|
@classmethod
|
|
def validate_status(cls, value: int):
|
|
"""校验状态：仅支持 0(正常)、1(禁用)"""
|
|
if value not in {0, 1}:
|
|
raise ValueError("状态仅支持 0(正常) 或 1(禁用)")
|
|
return value
|
|
|
|
@field_validator("name")
|
|
@classmethod
|
|
def validate_name(cls, value: str):
|
|
"""校验角色名称：不能为空"""
|
|
v = value.strip()
|
|
if not v:
|
|
raise ValueError("角色名称不能为空")
|
|
return v
|
|
|
|
|
|
class RolePermissionSettingSchema(BaseModel):
|
|
"""角色权限配置模型
|
|
"""
|
|
|
|
data_scope: int = Field(
|
|
default=1,
|
|
ge=1,
|
|
le=3,
|
|
description="数据权限范围(1:仅本人 2:本部门及以下 3:全部)",
|
|
)
|
|
role_ids: list[int] = Field(default_factory=list, description="角色ID列表")
|
|
menu_ids: list[int] = Field(default_factory=list, description="菜单ID列表")
|
|
dept_ids: list[int] = Field(default_factory=list, description="部门ID列表")
|
|
|
|
@model_validator(mode="after")
|
|
def validate_fields(self):
|
|
"""校验角色权限配置字段（数据范围与关联 ID 等）。
|
|
|
|
返回:
|
|
- RolePermissionSettingSchema: 通过 `role_permission_request_validator` 校验后的同一实例。
|
|
"""
|
|
return role_permission_request_validator(self)
|
|
|
|
|
|
class RoleUpdateSchema(RoleCreateSchema):
|
|
"""角色更新模型
|
|
"""
|
|
|
|
|
|
class RoleOutSchema(RoleCreateSchema, BaseSchema, UserBySchema):
|
|
"""角色信息响应模型
|
|
"""
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
menus: list[MenuOutSchema] = Field(default_factory=list, description="角色菜单列表")
|
|
depts: list[DeptOutSchema] = Field(default_factory=list, description="角色部门列表")
|
|
|
|
|
|
class RoleQueryParam(BaseQueryParam, UserByQueryParam):
|
|
"""角色管理查询参数
|
|
"""
|
|
|
|
name: str | None = Field(None, description="角色名称", json_schema_extra={"q": "like"})
|
|
code: str | None = Field(None, description="角色编码", json_schema_extra={"q": "eq"})
|
|
status: int | None = Field(None, description="状态(0:启动 1:停用)", json_schema_extra={"q": "eq"})
|