from typing import Optional
from pydantic import BaseModel


class CreateVehicle(BaseModel):
    Model: str
    Year: int
    LicensePlate: str
    Mileage: int
    Capacity: int


class OutputVehicle(BaseModel):
    Id: int
    Model: str
    Year: int
    LicensePlate: str
    Mileage: int
    CurrentLocation: Optional[list[str]] = None
    Fuel: Optional[int] = 0
    Type: str
    DriverHistory: Optional[list[int]] = None
    AssignedDriver: Optional[int] = None
    Capacity: int
    Status: str


class UpdateVehicle(OutputVehicle):
    pass


class VehicleLocation(BaseModel):
    CurrentLocation: list[str]