shared.py

"""
Shared data models and types for the GM Batch Processing POC.
"""
from dataclasses import dataclass
from typing import Optional, List
from enum import Enum


class CommandType(str, Enum):
    """Types of commands that can be sent to vehicles or back-office systems."""
    VEHICLE_COMMAND = "vehicle_command"  # Commands that connect to vehicle (slower, ~300-600/min)
    BACKOFFICE_UPDATE = "backoffice_update"  # Database updates (faster, ~1000s/min)
    PROFILE_FIX = "profile_fix"


class ProcessingStatus(str, Enum):
    """Status of VIN processing."""
    PENDING = "pending"
    PROCESSING = "processing"
    COMPLETED = "completed"
    FAILED = "failed"
    NO_CONSENT = "no_consent"


@dataclass
class VINData:
    """Data structure for a single VIN to be processed."""
    vin: str
    command: str
    command_type: str  # CommandType value (e.g., "vehicle_command", "backoffice_update")
    batch_id: str
    additional_params: Optional[dict] = None


@dataclass
class ConsentResult:
    """Result from consent check activity."""
    vin: str
    has_consent: bool
    reason: Optional[str] = None


@dataclass
class PublishResult:
    """Result from publishing to EventHub."""
    vin: str
    message_id: str
    topic: str
    success: bool


@dataclass
class DomainResponse:
    """Response received from domain via EventHub."""
    vin: str
    message_id: str
    status: str  # ProcessingStatus value (e.g., "completed", "failed")
    details: Optional[dict] = None


@dataclass
class VINProcessingResult:
    """Final result of processing a single VIN."""
    vin: str
    status: str  # ProcessingStatus value (e.g., "completed", "failed", "no_consent")
    message: Optional[str] = None
    details: Optional[dict] = None


@dataclass
class BatchProcessingResult:
    """Final result of processing an entire batch."""
    batch_id: str
    total_vins_processed: int
    successful: int
    failed: int
    duration_seconds: float