Add Fetch API to Python SDK

Author: jamesliupenn

dimo/api/__init__.py

@@ -1,6 +1,7 @@
 from .auth import Auth
 from .attestation import Attestation
 from .device_definitions import DeviceDefinitions
+from .fetch import Fetch
 from .token_exchange import TokenExchange
 from .trips import Trips
 from .valuations import Valuations
@@ -9,6 +10,7 @@
     "Auth",
     "Attestation",
     "DeviceDefinitions",
+    "Fetch",
     "TokenExchange",
     "Trips",
     "Valuations",

dimo/api/fetch.py

@@ -0,0 +1,123 @@
+from dimo.errors import check_type, check_optional_type
+
+
+class Fetch:
+
+    def __init__(self, request_method, get_auth_headers):
+        self._request = request_method
+        self._get_auth_headers = get_auth_headers
+
+    def _build_params(self, **kwargs):
+        params = {}
+        for key, value in kwargs.items():
+            if value is not None:
+                params[key] = value
+        return params
+
+    def get_index_keys(
+        self,
+        vehicle_jwt: str,
+        token_id: int,
+        after=None,
+        before=None,
+        id=None,
+        limit=None,
+        producer=None,
+        source=None,
+        type=None,
+    ) -> dict:
+        check_type("vehicle_jwt", vehicle_jwt, str)
+        check_type("token_id", token_id, int)
+        params = self._build_params(
+            after=after, before=before, id=id, limit=limit,
+            producer=producer, source=source, type=type,
+        )
+        url = f"/v1/vehicle/index-keys/{token_id}"
+        return self._request(
+            "GET",
+            "Fetch",
+            url,
+            params=params,
+            headers=self._get_auth_headers(vehicle_jwt),
+        )
+
+    def get_latest_index_key(
+        self,
+        vehicle_jwt: str,
+        token_id: int,
+        after=None,
+        before=None,
+        id=None,
+        limit=None,
+        producer=None,
+        source=None,
+        type=None,
+    ) -> dict:
+        check_type("vehicle_jwt", vehicle_jwt, str)
+        check_type("token_id", token_id, int)
+        params = self._build_params(
+            after=after, before=before, id=id, limit=limit,
+            producer=producer, source=source, type=type,
+        )
+        url = f"/v1/vehicle/latest-index-key/{token_id}"
+        return self._request(
+            "GET",
+            "Fetch",
+            url,
+            params=params,
+            headers=self._get_auth_headers(vehicle_jwt),
+        )
+
+    def get_latest_object(
+        self,
+        vehicle_jwt: str,
+        token_id: int,
+        after=None,
+        before=None,
+        id=None,
+        limit=None,
+        producer=None,
+        source=None,
+        type=None,
+    ) -> dict:
+        check_type("vehicle_jwt", vehicle_jwt, str)
+        check_type("token_id", token_id, int)
+        params = self._build_params(
+            after=after, before=before, id=id, limit=limit,
+            producer=producer, source=source, type=type,
+        )
+        url = f"/v1/vehicle/latest-object/{token_id}"
+        return self._request(
+            "GET",
+            "Fetch",
+            url,
+            params=params,
+            headers=self._get_auth_headers(vehicle_jwt),
+        )
+
+    def get_objects(
+        self,
+        vehicle_jwt: str,
+        token_id: int,
+        after=None,
+        before=None,
+        id=None,
+        limit=None,
+        producer=None,
+        source=None,
+        type=None,
+    ) -> dict:
+        check_type("vehicle_jwt", vehicle_jwt, str)
+        check_type("token_id", token_id, int)
+        params = self._build_params(
+            after=after, before=before, id=id, limit=limit,
+            producer=producer, source=source, type=type,
+        )
+        url = f"/v1/vehicle/objects/{token_id}"
+        return self._request(
+            "GET",
+            "Fetch",
+            url,
+            params=params,
+            headers=self._get_auth_headers(vehicle_jwt),
+        )

dimo/dimo.py

@@ -5,6 +5,7 @@
 from .api.conversations import Conversations
 from .api.device_definitions import DeviceDefinitions
 from .api.token_exchange import TokenExchange
+from .api.fetch import Fetch
 from .api.trips import Trips
 from .api.valuations import Valuations
 from .api.vehicle_triggers import VehicleTriggers
@@ -78,6 +79,7 @@ def __getattr__(self, name: str) -> Any:
             "auth": (Auth, ("request", "_get_auth_headers", "env", "self")),
             "conversations": (Conversations, ("request", "_get_auth_headers", "_get_full_path", "session")),
             "device_definitions": (DeviceDefinitions, ("request", "_get_auth_headers")),
+            "fetch": (Fetch, ("request", "_get_auth_headers")),
             "token_exchange": (
                 TokenExchange,
                 ("request", "_get_auth_headers", "identity", "self"),

dimo/environments.py

@@ -5,6 +5,7 @@
         "Conversations": "https://conversations-api.dimo.zone",
         "Identity": "https://identity-api.dimo.zone/query",
         "DeviceDefinitions": "https://device-definitions-api.dimo.zone",
+        "Fetch": "https://fetch-api.dimo.zone",
         "Telemetry": "https://telemetry-api.dimo.zone/query",
         "TokenExchange": "https://token-exchange-api.dimo.zone",
         "Trips": "https://trips-api.dimo.zone",
@@ -18,6 +19,7 @@
         "Conversations": "https://conversations-api.dev.dimo.zone",
         "Identity": "https://identity-api.dev.dimo.zone/query",
         "DeviceDefinitions": "https://device-definitions-api.dev.dimo.zone",
+        "Fetch": "https://fetch-api.dev.dimo.zone",
         "Telemetry": "https://telemetry-api.dev.dimo.zone/query",
         "TokenExchange": "https://token-exchange-api.dev.dimo.zone",
         "Trips": "https://trips-api.dev.dimo.zone",