Agent Zero
64 lines
1.7 KiB
Python
64 lines
1.7 KiB
Python
"""Pydantic schemas for EquipmentGroup (Bundles)."""
|
|
|
|
from __future__ import annotations
|
|
from datetime import datetime
|
|
from typing import TYPE_CHECKING
|
|
from pydantic import BaseModel, Field
|
|
|
|
if TYPE_CHECKING:
|
|
from app.schemas.equipment import LocationRef
|
|
|
|
|
|
class EquipmentGroupItem(BaseModel):
|
|
"""An item inside a bundle with quantity."""
|
|
equipment_id: str
|
|
name: str | None = None
|
|
quantity: float = 1.0
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class EquipmentGroupCreateRequest(BaseModel):
|
|
"""Request body for creating a new equipment group/bundle."""
|
|
name: str = Field(..., max_length=255)
|
|
description: str | None = None
|
|
daily_rate: float | None = None
|
|
default_location_id: str | None = None
|
|
items: list[EquipmentGroupItem] = []
|
|
|
|
|
|
class EquipmentGroupUpdateRequest(BaseModel):
|
|
"""Request body for updating an equipment group."""
|
|
name: str | None = Field(None, max_length=255)
|
|
description: str | None = None
|
|
daily_rate: float | None = None
|
|
default_location_id: str | None = None
|
|
items: list[EquipmentGroupItem] | None = None
|
|
|
|
|
|
class EquipmentGroupResponse(BaseModel):
|
|
"""Public equipment group representation."""
|
|
id: str
|
|
account_id: str
|
|
name: str
|
|
description: str | None = None
|
|
daily_rate: float | None = None
|
|
default_location: LocationRef | None = None
|
|
items: list[EquipmentGroupItem] = []
|
|
created_at: str
|
|
updated_at: str
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class EquipmentGroupListResponse(BaseModel):
|
|
"""Paginated list of equipment groups."""
|
|
items: list[EquipmentGroupResponse]
|
|
total: int
|
|
page: int
|
|
size: int
|
|
|
|
|
|
# Need LocationRef for EquipmentGroupResponse
|
|
from app.schemas.equipment import LocationRef # noqa: E402, F811
|