tesla-fleet-api 0.7.0__tar.gz → 0.7.2__tar.gz

{tesla_fleet_api-0.7.0/tesla_fleet_api.egg-info → tesla_fleet_api-0.7.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tesla_fleet_api
-Version: 0.7.0
+Version: 0.7.2
 Summary: Tesla Fleet API library for Python
 Home-page: https://github.com/Teslemetry/tesla_fleet_api
 Author: Brett Adams

{tesla_fleet_api-0.7.0 → tesla_fleet_api-0.7.2}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
 
 setuptools.setup(
     name="tesla_fleet_api",
-    version="0.7.0",
+    version="0.7.2",
     author="Brett Adams",
     author_email="hello@teslemetry.com",
     description="Tesla Fleet API library for Python",

{tesla_fleet_api-0.7.0 → tesla_fleet_api-0.7.2}/tesla_fleet_api/__init__.py

@@ -15,6 +15,7 @@ from .vehiclespecific import VehicleSpecific
 __all__ = [
     "TeslaFleetApi",
     "TeslaFleetOAuth",
+    "TeslaFleetOpenSource",
     "Teslemetry",
     "Tessie",
     "Charging",

{tesla_fleet_api-0.7.0 → tesla_fleet_api-0.7.2}/tesla_fleet_api/const.py

@@ -3,7 +3,7 @@
 from enum import Enum
 import logging
 
-VERSION = "0.7.0"
+VERSION = "0.7.2"
 LOGGER = logging.getLogger(__package__)
 SERVERS = {
     "na": "https://fleet-api.prd.na.vn.cloud.tesla.com",

{tesla_fleet_api-0.7.0 → tesla_fleet_api-0.7.2}/tesla_fleet_api/exceptions.py

@@ -363,7 +363,12 @@ async def raise_for_status(resp: aiohttp.ClientResponse) -> None:
     elif resp.status == 424:
         raise InvalidResponse(data)
     elif resp.status == 429:
-        raise RateLimited(data)
+        raise RateLimited(
+            {
+                "reset": resp.headers.get("RateLimit-Reset"),
+                "after": resp.headers.get("Retry-After"),
+            }
+        )
     elif resp.status == 451:
         raise ResourceUnavailableForLegalReasons(data)
     elif resp.status == 499:

{tesla_fleet_api-0.7.0 → tesla_fleet_api-0.7.2}/tesla_fleet_api/teslafleetapi.py

@@ -1,7 +1,7 @@
 """Tesla Fleet API for Python."""
 
 from json import dumps
-from typing import Any
+from typing import Any, Awaitable
 import aiohttp
 
 from .exceptions import raise_for_status, InvalidRegion, LibraryError, ResponseError
@@ -22,6 +22,7 @@ class TeslaFleetApi:
     server: str | None = None
     session: aiohttp.ClientSession
     headers: dict[str, str]
+    refresh_hook: Awaitable | None
 
     def __init__(
         self,
@@ -34,11 +35,13 @@ class TeslaFleetApi:
         partner_scope: bool = True,
         user_scope: bool = True,
         vehicle_scope: bool = True,
+        refresh_hook: Awaitable | None = None,
     ):
         """Initialize the Tesla Fleet API."""
 
         self.session = session
         self.access_token = access_token
+        self.refresh_hook = refresh_hook
 
         if server is not None:
             self.server = server
@@ -90,6 +93,11 @@ class TeslaFleetApi:
         if method == Method.GET and json is not None:
             raise ValueError("GET requests cannot have a body.")
 
+        # Call a pre-request hook if provided
+        if self.refresh_hook is not None:
+            if access_token := await self.refresh_hook():
+                self.access_token = access_token
+
         LOGGER.debug("Sending request to %s", path)
 
         # Remove None values from params and json
@@ -114,8 +122,16 @@ class TeslaFleetApi:
             LOGGER.debug("Response Status: %s", resp.status)
             if "x-txid" in resp.headers:
                 LOGGER.debug("Response TXID: %s", resp.headers["x-txid"])
+            if "RateLimit-Reset" in resp.headers:
+                LOGGER.debug(
+                    "Rate limit reset: %s", resp.headers.get("RateLimit-Reset")
+                )
+            if "Retry-After" in resp.headers:
+                LOGGER.debug("Retry after: %s", resp.headers.get("Retry-After"))
+
             if not resp.ok:
                 await raise_for_status(resp)
+
             if not resp.content_type.lower().startswith("application/json"):
                 LOGGER.debug("Response type is: %s", resp.content_type)
                 raise ResponseError(status=resp.status, data=await resp.text())

{tesla_fleet_api-0.7.0 → tesla_fleet_api-0.7.2/tesla_fleet_api.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tesla_fleet_api
-Version: 0.7.0
+Version: 0.7.2
 Summary: Tesla Fleet API library for Python
 Home-page: https://github.com/Teslemetry/tesla_fleet_api
 Author: Brett Adams