from pydantic import BaseModel, Field
from datetime import datetime
from schemas.user import ShowDriver

class CreateFuelingTask(BaseModel):
    VehicleId: int = Field(...)
    Description: str = Field(...)
    Date: datetime = Field(...)
    Cost: int = Field(...)
    FuelRefilled: int = Field(...)
    GasStationName: str = Field(...)
    ImageBefore: bytearray = Field(...)
    ImageAfter: bytearray = Field(...)
    model_config={
        "arbitrary_types_allowed": True
    }

class OutputFuelingTask(BaseModel):
    VehicleId: int = Field(...)
    Description: str = Field(...)
    Date: datetime = Field(...)
    Cost: int = Field(...)
    FuelRefilled: int = Field(...)
    GasStationName: str = Field(...)
    ImageBefore: bytearray = Field(...)
    ImageAfter: bytearray = Field(...)
    Driver: ShowDriver | None
    model_config={
        "arbitrary_types_allowed": True
    }