import uuid
from datetime import datetime
from typing import Optional

from pydantic import BaseModel, ConfigDict


class PrecautionCreate(BaseModel):
    site_name: str
    title: str
    content: str
    level: Optional[str] = None


class PrecautionUpdate(BaseModel):
    site_name: Optional[str] = None
    title: Optional[str] = None
    content: Optional[str] = None
    level: Optional[str] = None


class PrecautionRead(BaseModel):
    id: uuid.UUID
    study_id: uuid.UUID
    site_name: str
    title: str
    content: str
    level: Optional[str]
    created_by: Optional[uuid.UUID]
    created_at: datetime
    updated_at: datetime

    model_config = ConfigDict(from_attributes=True)