108 lines
3.2 KiB
Python
108 lines
3.2 KiB
Python
|
"""Pydantic schemas for Vehicles and VehicleAssignments."""
|
||
|
|
|
||
|
|
from pydantic import BaseModel, Field
|
||
|
|
|
||
|
|
|
||
|
|
class VehicleAssignmentResponse(BaseModel):
|
||
|
|
"""Public vehicle assignment representation."""
|
||
|
|
id: str
|
||
|
|
vehicle_id: str
|
||
|
|
project_id: str | None = None
|
||
|
|
start_date: str
|
||
|
|
end_date: str
|
||
|
|
status: str
|
||
|
|
notes: str | None = None
|
||
|
|
created_at: str
|
||
|
|
updated_at: str
|
||
|
|
|
||
|
|
model_config = {"from_attributes": True}
|
||
|
|
|
||
|
|
|
||
|
|
class VehicleCreateRequest(BaseModel):
|
||
|
|
"""Request body for creating a new vehicle."""
|
||
|
|
name: str = Field(..., max_length=255)
|
||
|
|
license_plate: str | None = Field(None, max_length=50)
|
||
|
|
brand: str | None = Field(None, max_length=100)
|
||
|
|
model: str | None = Field(None, max_length=100)
|
||
|
|
year: int | None = None
|
||
|
|
color: str | None = Field(None, max_length=50)
|
||
|
|
vehicle_type: str | None = Field(None, max_length=50)
|
||
|
|
payload_capacity_kg: float | None = None
|
||
|
|
load_volume_m3: float | None = None
|
||
|
|
fuel_type: str | None = Field(None, max_length=30)
|
||
|
|
is_active: bool = True
|
||
|
|
notes: str | None = None
|
||
|
|
|
||
|
|
|
||
|
|
class VehicleUpdateRequest(BaseModel):
|
||
|
|
"""Request body for updating a vehicle."""
|
||
|
|
name: str | None = Field(None, max_length=255)
|
||
|
|
license_plate: str | None = Field(None, max_length=50)
|
||
|
|
brand: str | None = Field(None, max_length=100)
|
||
|
|
model: str | None = Field(None, max_length=100)
|
||
|
|
year: int | None = None
|
||
|
|
color: str | None = Field(None, max_length=50)
|
||
|
|
vehicle_type: str | None = Field(None, max_length=50)
|
||
|
|
payload_capacity_kg: float | None = None
|
||
|
|
load_volume_m3: float | None = None
|
||
|
|
fuel_type: str | None = Field(None, max_length=30)
|
||
|
|
is_active: bool | None = None
|
||
|
|
notes: str | None = None
|
||
|
|
|
||
|
|
|
||
|
|
class VehicleResponse(BaseModel):
|
||
|
|
"""Public vehicle representation."""
|
||
|
|
id: str
|
||
|
|
account_id: str
|
||
|
|
name: str
|
||
|
|
license_plate: str | None = None
|
||
|
|
brand: str | None = None
|
||
|
|
model: str | None = None
|
||
|
|
year: int | None = None
|
||
|
|
color: str | None = None
|
||
|
|
vehicle_type: str | None = None
|
||
|
|
payload_capacity_kg: float | None = None
|
||
|
|
load_volume_m3: float | None = None
|
||
|
|
fuel_type: str | None = None
|
||
|
|
is_active: bool
|
||
|
|
notes: str | None = None
|
||
|
|
assignments: list[VehicleAssignmentResponse] = []
|
||
|
|
created_at: str
|
||
|
|
updated_at: str
|
||
|
|
|
||
|
|
model_config = {"from_attributes": True}
|
||
|
|
|
||
|
|
|
||
|
|
class VehicleListResponse(BaseModel):
|
||
|
|
"""Paginated list of vehicles."""
|
||
|
|
items: list[VehicleResponse]
|
||
|
|
total: int
|
||
|
|
page: int
|
||
|
|
size: int
|
||
|
|
|
||
|
|
|
||
|
|
class VehicleAssignmentCreateRequest(BaseModel):
|
||
|
|
"""Request body for creating a vehicle assignment."""
|
||
|
|
vehicle_id: str
|
||
|
|
project_id: str | None = None
|
||
|
|
start_date: str = Field(..., description="ISO datetime")
|
||
|
|
end_date: str = Field(..., description="ISO datetime")
|
||
|
|
status: str = Field("assigned", max_length=20)
|
||
|
|
notes: str | None = None
|
||
|
|
|
||
|
|
|
||
|
|
class VehicleAssignmentUpdateRequest(BaseModel):
|
||
|
|
"""Request body for updating a vehicle assignment."""
|
||
|
|
start_date: str | None = None
|
||
|
|
end_date: str | None = None
|
||
|
|
status: str | None = Field(None, max_length=20)
|
||
|
|
notes: str | None = None
|
||
|
|
|
||
|
|
|
||
|
|
class VehicleAssignmentListResponse(BaseModel):
|
||
|
|
"""Paginated list of vehicle assignments."""
|
||
|
|
items: list[VehicleAssignmentResponse]
|
||
|
|
total: int
|
||
|
|
page: int
|
||
|
|
size: int
|