Simon
117 lines
3.7 KiB
Python
117 lines
3.7 KiB
Python
# =============================================================================
|
||
# 企微IT智能服务台 — 员工 API
|
||
# =============================================================================
|
||
# 说明：提供员工相关的管理接口
|
||
# 接口列表：
|
||
# PUT /api/employees/{employee_id}/it-level — 更新员工IT技能等级
|
||
# =============================================================================
|
||
|
||
from typing import Optional
|
||
|
||
from fastapi import APIRouter, HTTPException
|
||
from pydantic import BaseModel, Field, field_validator
|
||
|
||
from app.utils.response import success_response
|
||
|
||
from app.schemas.employee import VALID_IT_LEVELS, VALID_LEVEL_SOURCES
|
||
|
||
# 创建路由器
|
||
router = APIRouter(prefix="/employees", tags=["员工管理"])
|
||
|
||
|
||
# --------------------------------------------------------------------------
|
||
# 请求 Schema
|
||
# --------------------------------------------------------------------------
|
||
|
||
class ItLevelUpdateRequest(BaseModel):
|
||
"""IT技能等级更新请求 Schema。"""
|
||
|
||
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
|
||
|
||
|
||
class ItLevelUpdateResponse(BaseModel):
|
||
"""IT技能等级更新响应 Schema。"""
|
||
|
||
employee_id: str
|
||
it_level: str
|
||
it_level_source: str
|
||
message: str
|
||
|
||
|
||
# --------------------------------------------------------------------------
|
||
# Mock 员工数据存储（IT 等级映射）
|
||
# --------------------------------------------------------------------------
|
||
|
||
# 简单的内存存储，key 为 employee_id，value 为 it_level
|
||
MOCK_EMPLOYEE_IT_LEVELS: dict = {
|
||
"emp-001": "silver",
|
||
"emp-002": "gold",
|
||
"emp-003": "bronze",
|
||
"emp-004": "platinum",
|
||
"emp-005": "diamond",
|
||
"emp-006": "silver",
|
||
"emp-007": "star",
|
||
"emp-008": "king",
|
||
}
|
||
|
||
|
||
# --------------------------------------------------------------------------
|
||
# API 接口
|
||
# --------------------------------------------------------------------------
|
||
|
||
@router.put("/{employee_id}/it-level")
|
||
async def update_employee_it_level(
|
||
employee_id: str,
|
||
request: ItLevelUpdateRequest,
|
||
):
|
||
"""更新员工IT技能等级。
|
||
|
||
坐席可以手动调整员工的IT技能等级，等级来源标记为 manual。
|
||
更新后等级立即生效，并记录来源以便追溯。
|
||
|
||
Args:
|
||
employee_id: 员工ID
|
||
request: 等级更新请求
|
||
|
||
Returns:
|
||
更新结果
|
||
"""
|
||
# 更新内存中的等级
|
||
old_level = MOCK_EMPLOYEE_IT_LEVELS.get(employee_id, "silver")
|
||
MOCK_EMPLOYEE_IT_LEVELS[employee_id] = request.it_level
|
||
|
||
# 构造等级名称映射
|
||
level_names = {
|
||
"bronze": "青铜",
|
||
"silver": "白银",
|
||
"gold": "黄金",
|
||
"platinum": "铂金",
|
||
"diamond": "钻石",
|
||
"star": "星耀",
|
||
"king": "王者",
|
||
}
|
||
|
||
return success_response(data=ItLevelUpdateResponse(
|
||
employee_id=employee_id,
|
||
it_level=request.it_level,
|
||
it_level_source=request.source,
|
||
message=f"IT等级已从 {level_names.get(old_level, old_level)} 调整为 {level_names.get(request.it_level, request.it_level)}",
|
||
).model_dump())
|