From 9f5b9de2c962ee5264732e0ba4342e7c041a057d Mon Sep 17 00:00:00 2001
From: Barrett <barrettkowalsky@gmail.com>
Date: Wed, 29 Oct 2025 11:14:03 -0400
Subject: [PATCH] Add complete Attestation API v2 endpoint coverage
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Expanded attestation.py to include all endpoints from the DIMO Attestation API documentation, providing full support for verifiable credentials including odometer statements, vehicle health reports, and position attestations.

- Updated create_vin_vc to use v2 endpoint
- Added create_odometer_statement for odometer reading VCs
- Added create_vehicle_health for health status verification
- Added create_vehicle_position for location attestations
- Added comprehensive docstrings to all methods
- Maintained backward compatibility with existing v1 endpoints

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>
---
 dimo/api/attestation.py | 118 ++++++++++++++++++++++++++++++++++++++--
 1 file changed, 114 insertions(+), 4 deletions(-)

diff --git a/dimo/api/attestation.py b/dimo/api/attestation.py
index 8d26ccd..a523949 100644
--- a/dimo/api/attestation.py
+++ b/dimo/api/attestation.py
@@ -1,4 +1,5 @@
-from dimo.errors import check_type
+from dimo.errors import check_type, check_optional_type
+from typing import Optional
 
 
 class Attestation:
@@ -7,22 +8,131 @@ def __init__(self, request_method, get_auth_headers):
         self._get_auth_headers = get_auth_headers
 
     def create_vin_vc(self, vehicle_jwt: str, token_id: int) -> dict:
+        """
+        Generate cryptographic proof of a vehicle's VIN or retrieve existing unexpired attestation.
+
+        Args:
+            vehicle_jwt (str): Authentication JWT token
+            token_id (int): Vehicle token identifier
+
+        Returns:
+            dict: Response containing vcUrl, vcQuery, and confirmation message
+        """
         check_type("vehicle_jwt", vehicle_jwt, str)
         check_type("token_id", token_id, int)
-        params = {"force": True}
-        url = f"/v1/vc/vin/{token_id}"
+        url = f"/v2/attestation/vin/{token_id}"
         return self._request(
             "POST",
             "Attestation",
             url,
-            params=params,
             headers=self._get_auth_headers(vehicle_jwt),
         )
 
     def create_pom_vc(self, vehicle_jwt: str, token_id: int) -> dict:
+        """
+        Create proof of movement verifiable credential (v1 API).
+
+        Args:
+            vehicle_jwt (str): Authentication JWT token
+            token_id (int): Vehicle token identifier
+
+        Returns:
+            dict: Response from the API
+        """
         check_type("vehicle_jwt", vehicle_jwt, str)
         check_type("token_id", token_id, int)
         url = f"/v1/vc/pom/{token_id}"
         return self._request(
             "POST", "Attestation", url, headers=self._get_auth_headers(vehicle_jwt)
         )
+
+    def create_odometer_statement(
+        self, vehicle_jwt: str, token_id: int, timestamp: Optional[str] = None
+    ) -> dict:
+        """
+        Produce verifiable odometer reading attestation.
+
+        Args:
+            vehicle_jwt (str): Authentication JWT token
+            token_id (int): Vehicle token identifier
+            timestamp (str, optional): Specific moment for reading (ISO 8601 format)
+
+        Returns:
+            dict: Success message directing to telemetry-api retrieval
+        """
+        check_type("vehicle_jwt", vehicle_jwt, str)
+        check_type("token_id", token_id, int)
+        check_optional_type("timestamp", timestamp, str)
+
+        url = f"/v2/attestation/odometer-statement/{token_id}"
+        data = {}
+        if timestamp:
+            data["timestamp"] = timestamp
+
+        return self._request(
+            "POST",
+            "Attestation",
+            url,
+            headers=self._get_auth_headers(vehicle_jwt),
+            data=data if data else None,
+        )
+
+    def create_vehicle_health(
+        self, vehicle_jwt: str, token_id: int, start_time: str, end_time: str
+    ) -> dict:
+        """
+        Generate health status verification for specified timeframe.
+
+        Args:
+            vehicle_jwt (str): Authentication JWT token
+            token_id (int): Vehicle token identifier
+            start_time (str): Report beginning (ISO 8601 format)
+            end_time (str): Report conclusion (ISO 8601 format)
+
+        Returns:
+            dict: Success message with telemetry-api retrieval instructions
+        """
+        check_type("vehicle_jwt", vehicle_jwt, str)
+        check_type("token_id", token_id, int)
+        check_type("start_time", start_time, str)
+        check_type("end_time", end_time, str)
+
+        url = f"/v2/attestation/vehicle-health/{token_id}"
+        data = {"startTime": start_time, "endTime": end_time}
+
+        return self._request(
+            "POST",
+            "Attestation",
+            url,
+            headers=self._get_auth_headers(vehicle_jwt),
+            data=data,
+        )
+
+    def create_vehicle_position(
+        self, vehicle_jwt: str, token_id: int, timestamp: str
+    ) -> dict:
+        """
+        Produce location verification at specified moment.
+
+        Args:
+            vehicle_jwt (str): Authentication JWT token
+            token_id (int): Vehicle token identifier
+            timestamp (str): Location snapshot timing (ISO 8601 format)
+
+        Returns:
+            dict: Success message with telemetry-api retrieval instructions
+        """
+        check_type("vehicle_jwt", vehicle_jwt, str)
+        check_type("token_id", token_id, int)
+        check_type("timestamp", timestamp, str)
+
+        url = f"/v2/attestation/vehicle-position/{token_id}"
+        data = {"timestamp": timestamp}
+
+        return self._request(
+            "POST",
+            "Attestation",
+            url,
+            headers=self._get_auth_headers(vehicle_jwt),
+            data=data,
+        )