59 lines
1.7 KiB
Python
59 lines
1.7 KiB
Python
"""值班排班 Pydantic schemas"""
|
||
from __future__ import annotations
|
||
|
||
from datetime import date
|
||
from typing import List, Literal, Optional
|
||
|
||
from pydantic import BaseModel, Field, field_validator
|
||
|
||
LinkStatus = Literal["open", "closed"]
|
||
SourceType = Literal["assign", "grab"]
|
||
|
||
|
||
class DutyAssignmentIn(BaseModel):
|
||
id: Optional[str] = None
|
||
source: SourceType = "assign"
|
||
userId: Optional[str] = None
|
||
personName: str = ""
|
||
|
||
|
||
class DutySlotSaveIn(BaseModel):
|
||
shiftType: str = Field(min_length=1, max_length=32)
|
||
capacity: int = Field(ge=1, le=99)
|
||
workPlans: List[str] = []
|
||
remark: Optional[str] = Field(default=None, max_length=500)
|
||
startTime: Optional[str] = Field(default=None, description="HH:mm")
|
||
endTime: Optional[str] = Field(default=None, description="HH:mm")
|
||
crossDay: bool = False
|
||
assignments: Optional[List[DutyAssignmentIn]] = None
|
||
|
||
@field_validator("workPlans", mode="before")
|
||
@classmethod
|
||
def _split_work_plans(cls, v):
|
||
if v is None:
|
||
return []
|
||
if isinstance(v, str):
|
||
return [p.strip() for p in v.replace("，", ",").split(",") if p.strip()]
|
||
return v
|
||
|
||
|
||
class DutyDaySaveBody(BaseModel):
|
||
dutyDate: date
|
||
slots: List[DutySlotSaveIn]
|
||
linkStatus: Optional[LinkStatus] = None
|
||
|
||
|
||
class DutyLinkBody(BaseModel):
|
||
linkStatus: LinkStatus
|
||
|
||
|
||
class DutyGrabBody(BaseModel):
|
||
shiftType: str = Field(min_length=1, max_length=32)
|
||
userId: str = Field(min_length=1, max_length=64)
|
||
|
||
|
||
class DutyDictEnsureBody(BaseModel):
|
||
dictType: Literal["duty_shift_type", "duty_work_plan", "duty_time_range"]
|
||
label: str = Field(min_length=1, max_length=100)
|
||
preferredValue: Optional[str] = Field(default=None, max_length=64)
|