Files
service-finder/backend/app/schemas/asset.py

73 lines
2.4 KiB
Python

from pydantic import BaseModel, ConfigDict, Field
from typing import Optional, Dict, Any, List
from uuid import UUID
from datetime import datetime
# --- KATALÓGUS SÉMÁK (Gyári adatok) ---
class AssetCatalogBase(BaseModel):
"""Alap katalógus adatok, amik a technikai dúsításból származnak."""
make: str
model: str
generation: Optional[str] = None
year_from: Optional[int] = None
year_to: Optional[int] = None
vehicle_class: Optional[str] = None
fuel_type: Optional[str] = None
engine_code: Optional[str] = None
# --- ÚJ TECHNIKAI MEZŐK (Robot v1.0.8 Smart Hunter adatai) ---
power_kw: Optional[int] = None
engine_capacity: Optional[int] = None
max_weight_kg: Optional[int] = None
axle_count: Optional[int] = None
body_type: Optional[str] = None
class AssetCatalogResponse(AssetCatalogBase):
"""Katalógus válasz séma azonosítóval és extra gyári adatokkal."""
id: int
factory_data: Optional[Dict[str, Any]] = None
# Pydantic v2 konfiguráció az ORM (SQLAlchemy) támogatáshoz
model_config = ConfigDict(from_attributes=True)
# --- JÁRMŰ SÉMÁK (Asset) ---
class AssetBase(BaseModel):
"""Jármű példány alapadatai (egyedi azonosítók)."""
vin: str = Field(..., min_length=17, max_length=17)
license_plate: str
name: Optional[str] = None
year_of_manufacture: Optional[int] = None
class AssetCreate(AssetBase):
"""Séma új jármű felvételéhez."""
make: str
model: str
vehicle_class: Optional[str] = "car"
fuel_type: Optional[str] = None
current_reading: Optional[int] = 0
class AssetResponse(AssetBase):
"""
Teljes jármű válasz séma.
Ez a séma tartalmazza a 'catalog' objektumot, amiben a dúsított műszaki adatok vannak.
"""
id: UUID
catalog_id: int
catalog: AssetCatalogResponse # Ez a pont kapcsolja össze a dúsított technikai adatokat
status: str
is_verified: bool
model_config = ConfigDict(from_attributes=True)
# --- DIGITÁLIS IKER (Full Profile) ---
class AssetFullProfile(BaseModel):
"""
Komplex jelentésekhez használt séma.
Összefogja az identitást, telemetriát, pénzügyeket és szerviztörténetet.
"""
identity: Dict[str, Any]
telemetry: Dict[str, Any]
financial_summary: Dict[str, Any]
service_history: List[Dict[str, Any]]
model_config = ConfigDict(from_attributes=True)