fund-tracer/backend/app/schemas/screenshot.py

"""Screenshot schemas."""

from datetime import datetime
from pydantic import BaseModel, ConfigDict


class ScreenshotResponse(BaseModel):
    model_config = ConfigDict(from_attributes=True)
    id: int
    case_id: int
    filename: str
    file_path: str
    status: str
    progress_step: str | None = None
    progress_percent: int = 0
    progress_detail: str | None = None
    started_at: datetime | None = None
    finished_at: datetime | None = None
    duration_ms: int | None = None
    error_message: str | None = None
    created_at: datetime


class ScreenshotListResponse(BaseModel):
    items: list[ScreenshotResponse]