from datetime import datetime
from uuid import UUID

from pydantic import BaseModel

from app.models.report import ReportType


class ReportCreate(BaseModel):
    report_type: ReportType
    include_summary: bool = True
    include_transactions: bool = True
    include_flow_chart: bool = True
    include_timeline: bool = True
    include_reasons: bool = True
    include_inquiry: bool = False
    include_screenshots: bool = False


class ReportOut(BaseModel):
    id: UUID
    case_id: UUID
    report_type: ReportType
    file_path: str
    version: int
    created_at: datetime

    model_config = {"from_attributes": True}


class ReportListOut(BaseModel):
    items: list[ReportOut]
    total: int