from datetime import datetime
from typing import Optional
from pydantic import BaseModel


class FlightState(BaseModel):
    flight_id: str
    status: str
    frames_processed: int
    frames_total: int
    current_frame: Optional[int] = None
    blocked: bool = False
    search_grid_size: Optional[int] = None
    created_at: datetime
    updated_at: datetime