63 lines
2.4 KiB
Python
63 lines
2.4 KiB
Python
from pydantic import BaseModel, ConfigDict, Field, field_validator
|
|
|
|
from app.core.base_schema import BaseQueryParam, BaseSchema, UserByQueryParam, UserBySchema
|
|
from app.core.validator import validate_required_code
|
|
|
|
|
|
class DeptCreateSchema(BaseModel):
|
|
"""部门创建模型"""
|
|
|
|
name: str = Field(..., min_length=1, max_length=64, description="部门名称")
|
|
order: int = Field(default=1, ge=0, description="显示顺序")
|
|
code: str = Field(..., min_length=2, max_length=64, description="部门编码")
|
|
parent_id: int | None = Field(default=None, ge=1, description="父部门ID")
|
|
status: int = Field(default=0, ge=0, le=1, description="状态(0:启动 1:停用)")
|
|
description: str | None = Field(default=None, max_length=255, description="备注")
|
|
|
|
@field_validator("name")
|
|
@classmethod
|
|
def validate_name(cls, value: str):
|
|
"""校验部门名称：不能为空"""
|
|
if not value or not value.strip():
|
|
raise ValueError("部门名称不能为空")
|
|
return value.strip()
|
|
|
|
@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
|
|
|
|
|
|
class DeptUpdateSchema(DeptCreateSchema):
|
|
"""部门更新模型"""
|
|
|
|
|
|
class DeptOutSchema(DeptCreateSchema, BaseSchema, UserBySchema):
|
|
"""部门详情响应模型（不含 children，用于详情和更新）"""
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
parent_name: str | None = Field(default=None, max_length=64, description="父部门名称")
|
|
|
|
|
|
class DeptTreeOutSchema(DeptOutSchema):
|
|
"""部门树形响应模型（含 children，用于树形列表）"""
|
|
|
|
children: list["DeptTreeOutSchema"] | None = Field(default=None, description="子部门列表")
|
|
|
|
|
|
class DeptQueryParam(BaseQueryParam, UserByQueryParam):
|
|
"""部门管理查询参数"""
|
|
|
|
name: str | None = Field(None, description="部门名称", json_schema_extra={"q": "like"})
|
|
status: int | None = Field(None, ge=0, le=1, description="状态(0:启动 1:停用)", json_schema_extra={"q": "eq"})
|