from datetime import datetime
from uuid import UUID

from pydantic import BaseModel

from app.models.case import CaseStatus


class CaseCreate(BaseModel):
    case_no: str
    title: str
    victim_name: str
    handler: str = ""


class CaseUpdate(BaseModel):
    title: str | None = None
    victim_name: str | None = None
    handler: str | None = None
    status: CaseStatus | None = None


class CaseOut(BaseModel):
    id: UUID
    case_no: str
    title: str
    victim_name: str
    handler: str
    status: CaseStatus
    image_count: int
    total_amount: float
    created_at: datetime
    updated_at: datetime

    model_config = {"from_attributes": True}


class CaseListOut(BaseModel):
    items: list[CaseOut]
    total: int