Cheng Zhou
53 lines
1.3 KiB
Python
53 lines
1.3 KiB
Python
import uuid
|
|
from datetime import date, datetime
|
|
from typing import Optional
|
|
|
|
from pydantic import BaseModel, ConfigDict, Field
|
|
|
|
from app.schemas.user import UserDisplay
|
|
|
|
|
|
class SiteDisplay(BaseModel):
|
|
id: uuid.UUID
|
|
name: Optional[str] = None
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
class MilestoneCreate(BaseModel):
|
|
type: str
|
|
name: Optional[str] = None
|
|
planned_date: Optional[date] = None
|
|
owner_id: Optional[uuid.UUID] = None
|
|
site_id: Optional[uuid.UUID] = None
|
|
notes: Optional[str] = None
|
|
status: str = Field(default="NOT_STARTED")
|
|
|
|
|
|
class MilestoneUpdate(BaseModel):
|
|
name: Optional[str] = None
|
|
planned_date: Optional[date] = None
|
|
actual_date: Optional[date] = None
|
|
status: Optional[str] = None
|
|
owner_id: Optional[uuid.UUID] = None
|
|
site_id: Optional[uuid.UUID] = None
|
|
notes: Optional[str] = None
|
|
|
|
|
|
class MilestoneRead(BaseModel):
|
|
id: uuid.UUID
|
|
study_id: uuid.UUID
|
|
type: str
|
|
name: str
|
|
planned_date: Optional[date]
|
|
actual_date: Optional[date]
|
|
status: str
|
|
owner_id: Optional[uuid.UUID]
|
|
site_id: Optional[uuid.UUID] = None
|
|
owner: Optional[UserDisplay] = None
|
|
site: Optional[SiteDisplay] = None
|
|
notes: Optional[str]
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|