Simon
119 lines
3.9 KiB
Python
119 lines
3.9 KiB
Python
# =============================================================================
|
||
# 企微IT智能服务台 — 员工 Pydantic Schema
|
||
# =============================================================================
|
||
# 说明：定义员工相关的请求/响应数据结构
|
||
# 包含：IT等级更新请求、员工响应 Schema
|
||
# =============================================================================
|
||
|
||
from datetime import datetime
|
||
from typing import Any, Dict, List, Optional
|
||
|
||
from pydantic import BaseModel, Field, field_validator
|
||
|
||
|
||
# --------------------------------------------------------------------------
|
||
# IT 等级合法值
|
||
# --------------------------------------------------------------------------
|
||
VALID_IT_LEVELS = {"bronze", "silver", "gold", "platinum", "diamond", "star", "king"}
|
||
|
||
# --------------------------------------------------------------------------
|
||
# 等级来源合法值
|
||
# --------------------------------------------------------------------------
|
||
VALID_LEVEL_SOURCES = {"system", "manual", "assessment"}
|
||
|
||
|
||
# --------------------------------------------------------------------------
|
||
# IT 等级更新请求 Schema
|
||
# --------------------------------------------------------------------------
|
||
class ItLevelUpdateRequest(BaseModel):
|
||
"""IT技能等级更新请求 Schema。
|
||
|
||
坐席手动调整员工IT技能等级时使用。
|
||
|
||
Attributes:
|
||
it_level: 新的IT技能等级
|
||
source: 等级来源（默认 manual）
|
||
"""
|
||
|
||
it_level: str = Field(..., description="IT技能等级: bronze/silver/gold/platinum/diamond/star/king")
|
||
source: str = Field(default="manual", description="等级来源: system/manual/assessment")
|
||
|
||
@field_validator("it_level")
|
||
@classmethod
|
||
def validate_it_level(cls, v: str) -> str:
|
||
"""校验IT等级值是否合法。"""
|
||
if v not in VALID_IT_LEVELS:
|
||
raise ValueError(f"无效的IT等级: {v}，合法值为: {VALID_IT_LEVELS}")
|
||
return v
|
||
|
||
@field_validator("source")
|
||
@classmethod
|
||
def validate_source(cls, v: str) -> str:
|
||
"""校验等级来源值是否合法。"""
|
||
if v not in VALID_LEVEL_SOURCES:
|
||
raise ValueError(f"无效的等级来源: {v}，合法值为: {VALID_LEVEL_SOURCES}")
|
||
return v
|
||
|
||
|
||
# --------------------------------------------------------------------------
|
||
# 员工响应 Schema（返回给前端的数据结构）
|
||
# --------------------------------------------------------------------------
|
||
class EmployeeResponse(BaseModel):
|
||
"""员工响应 Schema。
|
||
|
||
返回给前端的员工数据结构。
|
||
使用 from_attributes=True 支持从 SQLAlchemy 模型直接转换。
|
||
|
||
Attributes:
|
||
id: 员工记录唯一标识
|
||
corp_id: 企业微信企业ID
|
||
employee_id: 企微员工UserID
|
||
name: 员工姓名
|
||
department: 部门
|
||
position: 岗位
|
||
mobile: 手机号
|
||
email: 邮箱
|
||
avatar: 头像URL
|
||
status: 激活状态
|
||
it_level: IT技能等级
|
||
it_level_source: 等级来源
|
||
notes: 坐席备注
|
||
last_login_at: 最后登录时间
|
||
created_at: 创建时间
|
||
updated_at: 更新时间
|
||
"""
|
||
|
||
id: str
|
||
corp_id: str
|
||
employee_id: str
|
||
name: str
|
||
department: str
|
||
position: str
|
||
mobile: str
|
||
email: str
|
||
avatar: str
|
||
status: int
|
||
it_level: str = "silver"
|
||
it_level_source: str = "system"
|
||
notes: Dict[str, Any] = Field(default_factory=dict, description="坐席备注")
|
||
last_login_at: Optional[datetime] = None
|
||
created_at: datetime
|
||
updated_at: datetime
|
||
|
||
model_config = {"from_attributes": True}
|
||
|
||
|
||
# --------------------------------------------------------------------------
|
||
# 员工列表响应 Schema
|
||
# --------------------------------------------------------------------------
|
||
class EmployeeListResponse(BaseModel):
|
||
"""员工列表响应 Schema。
|
||
|
||
Attributes:
|
||
items: 员工列表
|
||
total: 总数
|
||
"""
|
||
|
||
items: List[EmployeeResponse]
|
||
total: int
|