56 lines
1.3 KiB
Python
56 lines
1.3 KiB
Python
from datetime import date, datetime
|
|
from uuid import UUID
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
LEAVE_TYPES = ["年假", "事假", "病假", "调休", "其他"]
|
|
|
|
|
|
class LeaveCreate(BaseModel):
|
|
manager_id: UUID
|
|
leave_type: str = Field(default="事假", pattern="^(年假|事假|病假|调休|其他)$")
|
|
start_date: date
|
|
end_date: date
|
|
reason: str = ""
|
|
|
|
|
|
class LeaveUpdate(BaseModel):
|
|
manager_id: UUID | None = None
|
|
leave_type: str | None = Field(default=None, pattern="^(年假|事假|病假|调休|其他)$")
|
|
start_date: date | None = None
|
|
end_date: date | None = None
|
|
reason: str | None = None
|
|
|
|
|
|
class LeaveOut(BaseModel):
|
|
id: UUID
|
|
manager_id: UUID
|
|
manager_name: str = ""
|
|
leave_type: str
|
|
start_date: date
|
|
end_date: date
|
|
days: int = 0
|
|
reason: str = ""
|
|
submitted_by: UUID
|
|
submitted_by_name: str = ""
|
|
created_at: datetime | None = None
|
|
updated_at: datetime | None = None
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class LeaveOverviewItem(BaseModel):
|
|
manager_id: UUID
|
|
manager_name: str
|
|
leave_type: str
|
|
start_date: date
|
|
end_date: date
|
|
days: int
|
|
|
|
|
|
class LeaveOverview(BaseModel):
|
|
week_start: str
|
|
week_end: str
|
|
total_on_leave: int
|
|
leave_list: list[LeaveOverviewItem]
|