Initial Commit

mvp-platform-services/vehicles/api/models/responses.py

@@ -0,0 +1,84 @@
+from pydantic import BaseModel
+from typing import List, Optional
+
+class MakeItem(BaseModel):
+    """Make item response model"""
+    id: int
+    name: str
+
+class ModelItem(BaseModel):
+    """Model item response model"""
+    id: int
+    name: str
+
+class TrimItem(BaseModel):
+    """Trim item response model"""
+    id: int
+    name: str
+
+class EngineItem(BaseModel):
+    """Engine item response model"""
+    id: int
+    name: str
+
+class TransmissionItem(BaseModel):
+    """Transmission item response model"""
+    name: str
+
+class MakesResponse(BaseModel):
+    """Makes response model"""
+    makes: List[MakeItem]
+
+class YearsResponse(BaseModel):
+    """Years response model"""
+    years: List[int]
+
+class ModelsResponse(BaseModel):
+    """Models response model"""
+    models: List[ModelItem]
+
+class TrimsResponse(BaseModel):
+    """Trims response model"""
+    trims: List[TrimItem]
+
+class EnginesResponse(BaseModel):
+    """Engines response model"""
+    engines: List[EngineItem]
+
+class TransmissionsResponse(BaseModel):
+    """Transmissions response model"""
+    transmissions: List[TransmissionItem]
+
+class VINDecodeResult(BaseModel):
+    """VIN decode result model"""
+    make: Optional[str] = None
+    model: Optional[str] = None
+    year: Optional[int] = None
+    trim_name: Optional[str] = None
+    engine_description: Optional[str] = None
+    transmission_description: Optional[str] = None
+    horsepower: Optional[float] = None
+    torque: Optional[float] = None
+    top_speed: Optional[float] = None
+    fuel: Optional[str] = None
+    confidence_score: Optional[float] = None
+    vehicle_type: Optional[str] = None
+
+class VINDecodeRequest(BaseModel):
+    """VIN decode request model"""
+    vin: str
+
+class VINDecodeResponse(BaseModel):
+    """VIN decode response model"""
+    vin: str
+    result: Optional[VINDecodeResult]
+    success: bool
+    error: Optional[str] = None
+
+class HealthResponse(BaseModel):
+    """Health check response model"""
+    status: str
+    database: str
+    cache: str
+    version: str
+    etl_last_run: Optional[str] = None