Changes Initial commit

ota_agent/models.py

@@ -0,0 +1,130 @@
+from __future__ import annotations
+
+from datetime import datetime, timezone
+from enum import Enum
+
+from pydantic import BaseModel, Field, ConfigDict
+
+
+def to_camel(value: str) -> str:
+    parts = value.split("_")
+    return parts[0] + "".join(part.capitalize() for part in parts[1:])
+
+
+class ApiModel(BaseModel):
+    model_config = ConfigDict(populate_by_name=True, alias_generator=to_camel)
+
+
+class AgentStatus(str, Enum):
+    IDLE = "IDLE"
+    HAS_UPDATE = "HAS_UPDATE"
+    WAIT_USER_CONFIRM = "WAIT_USER_CONFIRM"
+    BACKING_UP_DATABASE = "BACKING_UP_DATABASE"
+    PULLING_IMAGE = "PULLING_IMAGE"
+    RESTARTING_SERVICE = "RESTARTING_SERVICE"
+    HEALTH_CHECKING = "HEALTH_CHECKING"
+    SUCCESS = "SUCCESS"
+    FAILED = "FAILED"
+    ROLLED_BACK = "ROLLED_BACK"
+
+
+class ComponentImages(ApiModel):
+    images: dict[str, str] = Field(default_factory=dict)
+
+    def to_env_mapping(self) -> dict[str, str]:
+        return {key: value for key, value in self.images.items() if value}
+
+
+class UpdateManifest(ApiModel):
+    release_version: str
+    release_notes: str = ""
+    components: ComponentImages = Field(default_factory=ComponentImages)
+    upgrade_mode: str = "manual_confirm"
+
+
+class AgentState(BaseModel):
+    vehicle_id: str
+    vin: str
+    current_release: str
+    status: AgentStatus = AgentStatus.IDLE
+    available_update: UpdateManifest | None = None
+    last_check_at: str | None = None
+    last_heartbeat_at: str | None = None
+    last_result: str | None = None
+    updated_at: str = Field(default_factory=lambda: utc_now())
+
+
+class HeartbeatPayload(ApiModel):
+    vehicle_id: str
+    vin: str
+    current_release: str
+    agent_status: str
+    target_release: str | None = None
+    last_result: str | None = None
+    images: dict[str, str] = Field(default_factory=dict)
+    backup_file: str | None = None
+    updated_at: str
+
+
+class UpdateCheckRequest(ApiModel):
+    vehicle_id: str
+    vin: str
+    current_release: str
+
+
+class UpdateCheckResponse(ApiModel):
+    has_update: bool = False
+    manifest: UpdateManifest | None = None
+    message: str = ""
+
+
+class ReportPayload(ApiModel):
+    vehicle_id: str
+    vin: str
+    current_release: str
+    target_release: str | None = None
+    agent_status: str
+    success: bool
+    message: str = ""
+    images: dict[str, str] = Field(default_factory=dict)
+    backup_file: str | None = None
+    updated_at: str = Field(default_factory=lambda: utc_now())
+
+
+class ConfirmUpgradeRequest(BaseModel):
+    confirmed_by: str = "android-app"
+
+
+class PostponeUpgradeRequest(BaseModel):
+    reason: str = "user_postpone"
+
+
+class LocalStatusResponse(BaseModel):
+    vehicle_id: str
+    vin: str
+    current_release: str
+    status: str
+    available_update: UpdateManifest | None = None
+    last_result: str | None = None
+    updated_at: str
+
+
+class CommandResult(BaseModel):
+    success: bool
+    stdout: str = ""
+    stderr: str = ""
+    returncode: int = 0
+
+
+class HealthcheckResult(BaseModel):
+    success: bool
+    detail: str
+
+
+class OperationResult(BaseModel):
+    success: bool
+    detail: str
+
+
+def utc_now() -> str:
+    return datetime.now(timezone.utc).isoformat()