tesla-fleet-api 0.5.16__py3-none-any.whl → 0.6.0__py3-none-any.whl

tesla_fleet_api/energy.py

@@ -1,11 +1,17 @@
-from typing import Any
+from __future__ import annotations
+from typing import Any, TYPE_CHECKING
 from .const import Method, EnergyOperationMode, EnergyExportMode
 from .energyspecific import EnergySpecific
 
+if TYPE_CHECKING:
+    from .teslafleetapi import TeslaFleetApi
+
 
 class Energy:
     """Class describing the Tesla Fleet API partner endpoints"""
 
+    parent: TeslaFleetApi
+
     def __init__(self, parent):
         self._request = parent._request
 

tesla_fleet_api/energyspecific.py

@@ -1,13 +1,20 @@
-from typing import Any
+from __future__ import annotations
+from typing import Any, TYPE_CHECKING
 from .const import EnergyExportMode, EnergyOperationMode
 
+if TYPE_CHECKING:
+    from .energy import Energy
+
 
 class EnergySpecific:
     """Class describing the Tesla Fleet API partner endpoints"""
 
+    _parent: Energy
+    energy_site_id: int
+
     def __init__(
         self,
-        parent,
+        parent: Energy,
         energy_site_id: int,
     ):
         self._parent = parent

tesla_fleet_api/exceptions.py

@@ -7,9 +7,10 @@ class TeslaFleetError(BaseException):
 
     message: str = "An unknown error has occurred."
     status: int | None = None
-    data: dict | None = None
+    data: dict | str | None = None
+    key: str | None = None
 
-    def __init__(self, data: dict | None = None, status: int | None = None):
+    def __init__(self, data: dict | str | None = None, status: int | None = None):
         LOGGER.debug(self.message)
         self.data = data
         self.status = status or self.status
@@ -20,6 +21,7 @@ class ResponseError(TeslaFleetError):
     """The response from the server was not JSON."""
 
     message = "The response from the server was not JSON."
+    data: str | None = None
 
 
 class InvalidCommand(TeslaFleetError):
@@ -375,5 +377,5 @@ async def raise_for_status(resp: aiohttp.ClientResponse) -> None:
     elif resp.status == 540:
         raise DeviceUnexpectedResponse(data)
     elif data is None:
-        raise ResponseError(status=resp.status)
+        raise ResponseError(status=resp.status, data=await resp.text())
     resp.raise_for_status()

tesla_fleet_api/teslafleetapi.py

@@ -1,8 +1,10 @@
-import aiohttp
+"""Tesla Fleet API for Python."""
 
 from json import dumps
-from .exceptions import raise_for_status, InvalidRegion, LibraryError, InvalidToken
 from typing import Any
+import aiohttp
+
+from .exceptions import raise_for_status, InvalidRegion, LibraryError, ResponseError
 from .const import SERVERS, Method, LOGGER, VERSION
 from .charging import Charging
 from .energy import Energy
@@ -15,10 +17,11 @@ from .vehicle import Vehicle
 class TeslaFleetApi:
     """Class describing the Tesla Fleet API."""
 
+    access_token: str | None = None
+    region: str | None = None
     server: str | None = None
     session: aiohttp.ClientSession
     headers: dict[str, str]
-    raise_for_status: bool
 
     def __init__(
         self,
@@ -26,7 +29,6 @@ class TeslaFleetApi:
         access_token: str | None = None,
         region: str | None = None,
         server: str | None = None,
-        raise_for_status: bool = True,
         charging_scope: bool = True,
         energy_scope: bool = True,
         partner_scope: bool = True,
@@ -37,7 +39,6 @@ class TeslaFleetApi:
 
         self.session = session
         self.access_token = access_token
-        self.raise_for_status = raise_for_status
 
         if server is not None:
             self.server = server
@@ -82,7 +83,7 @@ class TeslaFleetApi:
         path: str,
         params: dict[str, Any] | None = None,
         json: dict[str, Any] | None = None,
-    ) -> dict[str, Any] | str:
+    ) -> dict[str, Any]:
         """Send a request to the Tesla Fleet API."""
 
         if not self.server:
@@ -113,22 +114,14 @@ class TeslaFleetApi:
             params=params,
         ) as resp:
             LOGGER.debug("Response Status: %s", resp.status)
-            if self.raise_for_status and not resp.ok:
+            if not resp.ok:
                 await raise_for_status(resp)
-            elif resp.status == 401 and resp.content_type != "application/json":
-                # Manufacture a response since Tesla doesn't provide a body for token expiration.
-                return {
-                    "response": None,
-                    "error": InvalidToken.key,
-                    "error_message": "The OAuth token has expired.",
-                }
-            if resp.content_type == "application/json":
-                data = await resp.json()
-                LOGGER.debug("Response JSON: %s", data)
-                return data
-
-            data = await resp.text()
-            LOGGER.debug("Response Text: %s", data)
+            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())
+
+            data = await resp.json()
+            LOGGER.debug("Response JSON: %s", data)
             return data
 
     async def status(self) -> str:

tesla_fleet_api/teslafleetoauth.py

@@ -21,7 +21,6 @@ class TeslaFleetOAuth(TeslaFleetApi):
         expires: int = 0,
         region: str | None = None,
         server: str | None = None,
-        raise_for_status: bool = True,
     ):
         self.client_id = client_id
         self.access_token = access_token
@@ -33,7 +32,6 @@ class TeslaFleetOAuth(TeslaFleetApi):
             access_token="",
             region=region,
             server=server,
-            raise_for_status=raise_for_status,
         )
 
     def get_login_url(
@@ -96,8 +94,8 @@ class TeslaFleetOAuth(TeslaFleetApi):
         method: Method,
         path: str,
         params: dict[str, Any] | None = None,
-        data: dict[str, Any] | None = None,
-    ) -> str | dict[str, Any]:
+        json: dict[str, Any] | None = None,
+    ) -> dict[str, Any]:
         """Send a request to the Tesla Fleet API."""
         await self.check_access_token()
-        return await super()._request(method, path, params, data)
+        return await super()._request(method, path, params, json)

tesla_fleet_api/teslemetry.py

@@ -13,14 +13,12 @@ class Teslemetry(TeslaFleetApi):
         self,
         session: aiohttp.ClientSession,
         access_token: str,
-        raise_for_status: bool = True,
     ):
         """Initialize the Teslemetry API."""
         super().__init__(
             session,
             access_token,
             server="https://api.teslemetry.com",
-            raise_for_status=raise_for_status,
             partner_scope=False,
             user_scope=False,
         )
@@ -52,10 +50,11 @@ class Teslemetry(TeslaFleetApi):
             LOGGER.debug("Using server %s", self.server)
         return resp
 
-    # TODO: type this properly, it probably should return something
-    async def find_server(self) -> None:
+    async def find_server(self) -> str:
         """Find the server URL for the Tesla Fleet API."""
         await self.metadata(True)
+        assert self.region
+        return self.region
 
     async def _request(
         self,
@@ -63,7 +62,7 @@ class Teslemetry(TeslaFleetApi):
         path: str,
         params: dict[str, Any] | None = None,
         json: dict[str, Any] | None = None,
-    ) -> str | dict[str, Any]:
+    ) -> dict[str, Any]:
         """Send a request to the Teslemetry API."""
         async with rate_limit:
             return await super()._request(method, path, params, json)

tesla_fleet_api/tessie.py

@@ -9,14 +9,12 @@ class Tessie(TeslaFleetApi):
         self,
         session: aiohttp.ClientSession,
         access_token: str,
-        raise_for_status: bool = True,
     ):
         """Initialize the Tessie API."""
         super().__init__(
             session,
             access_token,
             server="https://api.tessie.com",
-            raise_for_status=raise_for_status,
             partner_scope=False,
             user_scope=False,
         )

tesla_fleet_api/vehicle.py

@@ -1,4 +1,5 @@
-from typing import Any, List
+from __future__ import annotations
+from typing import Any, List, TYPE_CHECKING
 from .const import (
     Method,
     Trunk,
@@ -13,17 +14,22 @@ from .const import (
 )
 from .vehiclespecific import VehicleSpecific
 
+if TYPE_CHECKING:
+    from .teslafleetapi import TeslaFleetApi
+
 
 class Vehicle:
     """Class describing the Tesla Fleet API vehicle endpoints and commands."""
 
-    def __init__(self, parent):
+    _parent: TeslaFleetApi
+
+    def __init__(self, parent: TeslaFleetApi):
         self._parent = parent
         self._request = parent._request
 
-    def specific(self, vehicle_tag: str | int) -> VehicleSpecific:
+    def specific(self, vin: str) -> VehicleSpecific:
         """Creates a class for each vehicle."""
-        return VehicleSpecific(self, vehicle_tag)
+        return VehicleSpecific(self, vin)
 
     def pre2021(self, vin: str) -> bool:
         """Checks if a vehicle is a pre-2021 model S or X."""
@@ -716,26 +722,6 @@ class Vehicle:
             {"routable_message": routable_message},
         )
 
-    async def subscriptions(
-        self, device_token: str, device_type: str
-    ) -> dict[str, Any]:
-        """Returns the list of vehicles for which this mobile device currently subscribes to push notifications."""
-        return await self._request(
-            Method.GET,
-            "api/1/subscriptions",
-            query={"device_token": device_token, "device_type": device_type},
-        )
-
-    async def subscriptions_set(
-        self, device_token: str, device_type: str
-    ) -> dict[str, Any]:
-        """Allows a mobile device to specify which vehicles to receive push notifications from."""
-        return await self._request(
-            Method.POST,
-            "api/1/subscriptions",
-            query={"device_token": device_token, "device_type": device_type},
-        )
-
     async def vehicle(self, vehicle_tag: str | int) -> dict[str, Any]:
         """Returns information about a vehicle."""
         return await self._request(Method.GET, f"api/1/vehicles/{vehicle_tag}")
@@ -743,7 +729,7 @@ class Vehicle:
     async def vehicle_data(
         self,
         vehicle_tag: str | int,
-        endpoints: List[VehicleDataEndpoint] | List[str] | None = None,
+        endpoints: List[VehicleDataEndpoint | str] | None = None,
     ) -> dict[str, Any]:
         """Makes a live call to the vehicle. This may return cached data if the vehicle is offline. For vehicles running firmware versions 2023.38+, location_data is required to fetch vehicle location. This will result in a location sharing icon to show on the vehicle UI."""
         endpoint_payload = ";".join(endpoints) if endpoints else None
@@ -753,37 +739,21 @@ class Vehicle:
             {"endpoints": endpoint_payload},
         )
 
-    async def vehicle_subscriptions(
-        self, device_token: str, device_type: DeviceType | str
-    ) -> dict[str, Any]:
-        """Returns the list of vehicles for which this mobile device currently subscribes to push notifications."""
-        return await self._request(
-            Method.GET,
-            "api/1/vehicle_subscriptions",
-            {"device_token": device_token, "device_type": device_type},
-        )
-
-    async def vehicle_subscriptions_set(
-        self, device_token: str, device_type: DeviceType | str
-    ) -> dict[str, Any]:
-        """Allows a mobile device to specify which vehicles to receive push notifications from."""
-        return await self._request(
-            Method.POST,
-            "api/1/vehicle_subscriptions",
-            params={"device_token": device_token, "device_type": device_type},
-        )
-
     async def wake_up(self, vehicle_tag: str | int) -> dict[str, Any]:
         """Wakes the vehicle from sleep, which is a state to minimize idle energy consumption."""
         return await self._request(Method.POST, f"api/1/vehicles/{vehicle_tag}/wake_up")
 
     async def warranty_details(self, vin: str | None) -> dict[str, Any]:
         """Returns warranty details."""
-        return await self._request(Method.GET, "api/1/dx/warranty/details", {vin: vin})
+        return await self._request(
+            Method.GET, "api/1/dx/warranty/details", {"vin": vin}
+        )
 
     async def fleet_status(self, vins: List[str]) -> dict[str, Any]:
         """Checks whether vehicles can accept Tesla commands protocol for the partner's public key"""
-        return await self._request(Method.GET, "api/1/vehicles/fleet_status", json=vins)
+        return await self._request(
+            Method.GET, "api/1/vehicles/fleet_status", json={"vins": vins}
+        )
 
     async def fleet_telemetry_config_create(
         self, config: dict[str, Any]

tesla_fleet_api/vehiclespecific.py

@@ -1,4 +1,5 @@
-from typing import Any
+from __future__ import annotations
+from typing import Any, TYPE_CHECKING
 from .const import (
     Trunk,
     ClimateKeeperMode,
@@ -9,11 +10,17 @@ from .const import (
     DeviceType,
 )
 
+if TYPE_CHECKING:
+    from .vehicle import Vehicle
+
 
 class VehicleSpecific:
     """Class describing the Tesla Fleet API vehicle endpoints and commands for a specific vehicle."""
 
-    def __init__(self, parent, vin: str | int | None = None):
+    _parent: Vehicle
+    vin: str
+
+    def __init__(self, parent: Vehicle, vin: str):
         self._parent = parent
         self.vin = vin
 
@@ -406,45 +413,17 @@ class VehicleSpecific:
         """Signed Commands is a generic endpoint replacing legacy commands."""
         return await self._parent.signed_command(self.vin, routable_message)
 
-    async def subscriptions(
-        self, device_token: str, device_type: str
-    ) -> dict[str, Any]:
-        """Returns the list of vehicles for which this mobile device currently subscribes to push notifications."""
-        return await self._parent.subscriptions(self.vin, device_token, device_type)
-
-    async def subscriptions_set(
-        self, device_token: str, device_type: str
-    ) -> dict[str, Any]:
-        """Allows a mobile device to specify which vehicles to receive push notifications from."""
-        return await self._parent.subscriptions_set(self.vin, device_token, device_type)
-
     async def vehicle(self) -> dict[str, Any]:
         """Returns information about a vehicle."""
         return await self._parent.vehicle(self.vin)
 
     async def vehicle_data(
         self,
-        endpoints: list[VehicleDataEndpoint] | str | None = None,
+        endpoints: list[VehicleDataEndpoint | str] | None = None,
     ) -> dict[str, Any]:
         """Makes a live call to the vehicle. This may return cached data if the vehicle is offline. For vehicles running firmware versions 2023.38+, location_data is required to fetch vehicle location. This will result in a location sharing icon to show on the vehicle UI."""
         return await self._parent.vehicle_data(self.vin, endpoints)
 
-    async def vehicle_subscriptions(
-        self, device_token: str, device_type: DeviceType | str
-    ) -> dict[str, Any]:
-        """Returns the list of vehicles for which this mobile device currently subscribes to push notifications."""
-        return await self._parent.vehicle_subscriptions(
-            self.vin, device_token, device_type
-        )
-
-    async def vehicle_subscriptions_set(
-        self, device_token: str, device_type: DeviceType | str
-    ) -> dict[str, Any]:
-        """Allows a mobile device to specify which vehicles to receive push notifications from."""
-        return await self._parent.vehicle_subscriptions_set(
-            self.vin, device_token, device_type
-        )
-
     async def wake_up(self) -> dict[str, Any]:
         """Wakes the vehicle from sleep, which is a state to minimize idle energy consumption."""
         return await self._parent.wake_up(self.vin)

{tesla_fleet_api-0.5.16.dist-info → tesla_fleet_api-0.6.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tesla_fleet_api
-Version: 0.5.16
+Version: 0.6.0
 Summary: Tesla Fleet API library for Python
 Home-page: https://github.com/Teslemetry/tesla_fleet_api
 Author: Brett Adams
@@ -40,7 +40,6 @@ async def main():
             access_token="<access_token>",
             session=session,
             region="na",
-            raise_for_status=True,
         )
 
         try:
@@ -71,7 +70,6 @@ async def main():
             refresh_token=auth["refresh_token"],
             expires=auth["expires"],
             region="na",
-            raise_for_status=True,
         )
         try:
             data = await api.vehicle.list()
@@ -108,7 +106,6 @@ async def main():
         api = Teslemetry(
             access_token="<access_token>",
             session=session,
-            raise_for_status=True,
         )
 
         try:
@@ -136,7 +133,6 @@ async def main():
         api = Tessie(
             access_token="<access_token>",
             session=session,
-            raise_for_status=True,
         )
 
         try:

tesla_fleet_api-0.6.0.dist-info/RECORD

@@ -0,0 +1,19 @@
+tesla_fleet_api/__init__.py,sha256=0MON9vh3AShIiX16FZ6NU3yZ7kyXFh5GxA0rY8CzVRM,584
+tesla_fleet_api/charging.py,sha256=N_mc8axrXj3iduqLj_jCt4Vx86tHqe3xqQT4R1R7HvU,1689
+tesla_fleet_api/const.py,sha256=-MYIW8BaV-B0w7dxx--sZyF7bZQslLsEtxZZ_vIYPy8,9278
+tesla_fleet_api/energy.py,sha256=lVcvrU7W2Nbd0L0j4wytzD8uV3wl0oMXN07Zp90K4t8,5426
+tesla_fleet_api/energyspecific.py,sha256=KWfCV609r8MvNAcg8d1PbrjVH_mrJDjzdi_WEnFiUGs,3885
+tesla_fleet_api/exceptions.py,sha256=BcvJyjY4qM2-Rf7ZMA94xPIN6ebKit89E7qattMmbr8,11018
+tesla_fleet_api/partner.py,sha256=1vIBUaxKLIfqcC0X6VXZN0dMAzj_CLNPUMjA6QVqZ1k,1223
+tesla_fleet_api/teslafleetapi.py,sha256=-hiZfEEdS8Sfj_yqW-woiD5XD5_Q2IcMubmkh7wdtdE,4826
+tesla_fleet_api/teslafleetoauth.py,sha256=OC6djQw2ieZR_rWK5HoInLScZOu5T5aMClbg5gqbnDg,3502
+tesla_fleet_api/teslemetry.py,sha256=07vY6y56slUXvyaQDGQSGW_VF98fQUgEh-xXrG_phw8,2000
+tesla_fleet_api/tessie.py,sha256=dZs85N2uVlqkjxB3ICRVnPyyJkl4tWDAb3CQqV4MqQc,2096
+tesla_fleet_api/user.py,sha256=TZE2oh-n5zrhKXmGRuiNL9voKVODD7rBhGE_IObYVGA,1179
+tesla_fleet_api/vehicle.py,sha256=EE8ZgL1QdOVfwPsShGeI9DZtsKcLKG7EcfBCnEugfH0,31411
+tesla_fleet_api/vehiclespecific.py,sha256=BhWaQGsNhg06gSLGqNEs2JQgolY3mNGtOWz478lofFE,20324
+tesla_fleet_api-0.6.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+tesla_fleet_api-0.6.0.dist-info/METADATA,sha256=Wmjs_Fx05K2qujHe4Jx_EhS8lVXjqosES1fvsK4TxUI,3821
+tesla_fleet_api-0.6.0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+tesla_fleet_api-0.6.0.dist-info/top_level.txt,sha256=jeNbog_1saXBFrGpom9WyPWmilxsyP3szL_G7JLWQfM,16
+tesla_fleet_api-0.6.0.dist-info/RECORD,,

tesla_fleet_api-0.5.16.dist-info/RECORD

@@ -1,19 +0,0 @@
-tesla_fleet_api/__init__.py,sha256=0MON9vh3AShIiX16FZ6NU3yZ7kyXFh5GxA0rY8CzVRM,584
-tesla_fleet_api/charging.py,sha256=N_mc8axrXj3iduqLj_jCt4Vx86tHqe3xqQT4R1R7HvU,1689
-tesla_fleet_api/const.py,sha256=-MYIW8BaV-B0w7dxx--sZyF7bZQslLsEtxZZ_vIYPy8,9278
-tesla_fleet_api/energy.py,sha256=kE-HDupzhgatIsizJoer1MAALP-wH6jjjGliiRQN0Os,5285
-tesla_fleet_api/energyspecific.py,sha256=kICxdeDoWR9JHlgjHvnmjJ1ErLOWJT8bCSESoXo9axU,3732
-tesla_fleet_api/exceptions.py,sha256=3QUqIqA2wBqr-dpatDtmz8nOnqKN-pJ4gnIy6PMDQPs,10927
-tesla_fleet_api/partner.py,sha256=1vIBUaxKLIfqcC0X6VXZN0dMAzj_CLNPUMjA6QVqZ1k,1223
-tesla_fleet_api/teslafleetapi.py,sha256=KYtfkqthYxWuKpctLbyAcE32sN5yMpJaCsMSWM7bX2E,5199
-tesla_fleet_api/teslafleetoauth.py,sha256=FfLnuqZMxF2HsZ5miLNtm3pRwKDoQKi2GD-oXgKxZgA,3594
-tesla_fleet_api/teslemetry.py,sha256=XOqm7SaakooJSNXEasJI_tRoCIyiFCD6n_PCx9U_D9g,2107
-tesla_fleet_api/tessie.py,sha256=Aso7pGcypG19Vgh8UBACJEdinbF5135e1aWdmrOxw7s,2182
-tesla_fleet_api/user.py,sha256=TZE2oh-n5zrhKXmGRuiNL9voKVODD7rBhGE_IObYVGA,1179
-tesla_fleet_api/vehicle.py,sha256=qM0lD4JnZOrlqHCLnXEpTQoVGUTiWr--tdTB8wKJrfA,32871
-tesla_fleet_api/vehiclespecific.py,sha256=Oaz5InhWwXuKgyoQSYWpLLsMPjZ2iF0bShlnDLxEk5k,21506
-tesla_fleet_api-0.5.16.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-tesla_fleet_api-0.5.16.dist-info/METADATA,sha256=Poyg10sjR6EfdB02U6m5fBWf6LrSzGBmHKv07JTir34,3962
-tesla_fleet_api-0.5.16.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-tesla_fleet_api-0.5.16.dist-info/top_level.txt,sha256=jeNbog_1saXBFrGpom9WyPWmilxsyP3szL_G7JLWQfM,16
-tesla_fleet_api-0.5.16.dist-info/RECORD,,