import uuid
from datetime import date, datetime

from pydantic import BaseModel, ConfigDict


class ProjectMilestoneRead(BaseModel):
    id: uuid.UUID
    study_id: uuid.UUID
    name: str
    planned_date: date | None
    adjusted_start_date: date | None
    adjusted_end_date: date | None
    actual_start_date: date | None
    actual_end_date: date | None
    status: str
    notes: str | None
    owner_name: str | None
    updated_at: datetime

    model_config = ConfigDict(from_attributes=True)


class ProjectMilestoneUpdate(BaseModel):
    adjusted_start_date: date | None = None
    adjusted_end_date: date | None = None
    actual_start_date: date | None = None
    actual_end_date: date | None = None
    notes: str | None = None