pydantic models for input and output? #39
Description
Is there a place somewhere to keep pydantic models of vroom input and output? I need them anyways so I thought I could share them. Looks something like this:
from pydantic import BaseModel
class Vehicle(BaseModel):
id: int
profile: Optional[str] = None
description: Optional[str] = None
start_index: Optional[int] = None
end_index: Optional[int] = None
capacity: Optional[Tuple[int]] = None
skills: Optional[Tuple[int]] = None
time_window: Optional[Tuple[int, int]] = None # timestamp or relative seconds
max_tasks: Optional[int] = None
class Matrix(BaseModel):
durations: List[List[int]]
class Job(BaseModel):
id: int
location_index: int
description: Optional[str] = None
service: Optional[int] = None
class VroomRequest(BaseModel):
vehicles: List[Vehicle]
matrices: Dict[str, Matrix]
jobs: List[Job]