tesla-fleet-api 0.7.2__py3-none-any.whl → 0.7.4__py3-none-any.whl

tesla_fleet_api/const.py

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

tesla_fleet_api/ratecalculator.py

@@ -0,0 +1,60 @@
+"""Rate Calculator helper"""
+
+import time
+
+DAY = 24 * 60 * 60
+
+
+class RateCalculator:
+    """Calculate the consumption and remaining rate of a rate limit."""
+
+    def __init__(
+        self,
+        limit: int,
+        period: int = 86400,
+        min_wait: int = 0,
+        max_wait: int | None = None,
+        factor: int = 5,
+    ) -> None:
+        """Initialize the rate calculator."""
+        self.limit: int = limit
+        self.period: int = period
+        self.history: list[int] = []
+        self.start = time.time()
+        self.min_wait = min_wait
+        self.max_wait = max_wait if max_wait is not None else period
+        self.factor = factor
+
+    def constrain(self, value: float) -> float:
+        """Constrain a value between min and max."""
+        return max(self.min_wait, min(self.max_wait, value))
+
+    def consume(self, timestamp: int | None = None) -> None:
+        """Consume a unit of the rate limit."""
+        now = timestamp or int(time.time() + 1)
+        self.history.append(now)
+
+    def calculate(self, timestamp: int | None = None) -> float:
+        """Return the ideal delay to avoid rate limiting."""
+
+        count = len(self.history)
+        if count == 0:
+            return self.min_wait
+
+        now = timestamp or int(time.time())
+
+        while self.history and self.history[0] < now - self.period:
+            self.history.pop(0)
+
+        remaining = self.limit - count
+
+        if remaining <= 0:
+            # The wait until a request is available
+            return self.constrain(self.history[abs(remaining)] + self.period - now)
+
+        return self.constrain(self.period / remaining / self.factor)
+
+    @property
+    def count(self) -> int:
+        """Return the number of requests in the current period."""
+        return len(self.history)

tesla_fleet_api/teslafleetapi.py

@@ -22,7 +22,7 @@ class TeslaFleetApi:
     server: str | None = None
     session: aiohttp.ClientSession
     headers: dict[str, str]
-    refresh_hook: Awaitable | None
+    refresh_hook: Awaitable | None = None
 
     def __init__(
         self,
@@ -114,7 +114,7 @@ class TeslaFleetApi:
             headers={
                 "Authorization": f"Bearer {self.access_token}",
                 "Content-Type": "application/json",
-                "X-Library": f"python tesla_fleet_api ${VERSION}",
+                "X-Library": f"python tesla_fleet_api {VERSION}",
             },
             json=json,
             params=params,

tesla_fleet_api/teslemetry.py

@@ -2,12 +2,18 @@ import aiohttp
 from aiolimiter import AsyncLimiter
 from typing import Any
 from .teslafleetapi import TeslaFleetApi
+from .charging import Charging
+from .energy import Energy
+from .partner import Partner
+from .user import User
+from .vehicle import Vehicle
+from .vehiclespecific import VehicleSpecific
+
 from .const import Method, LOGGER, Scope
 
 # Rate limit should be global, even if multiple instances are created
 rate_limit = AsyncLimiter(5, 10)
 
-
 class Teslemetry(TeslaFleetApi):
     def __init__(
         self,
@@ -16,11 +22,11 @@ class Teslemetry(TeslaFleetApi):
     ):
         """Initialize the Teslemetry API."""
         super().__init__(
-            session,
-            access_token,
+            session=session,
+            access_token=access_token,
             server="https://api.teslemetry.com",
-            partner_scope=False,
             user_scope=False,
+            partner_scope=False
         )
         self.rate_limit = rate_limit
 
@@ -38,6 +44,14 @@ class Teslemetry(TeslaFleetApi):
             "api/test",
         )
 
+    async def userdata(self) -> dict[str, Any]:
+        """Get userdata."""
+        resp = await self._request(
+            Method.GET,
+            "api/userdata",
+        )
+        return resp
+
     async def metadata(self, update_region=True) -> dict[str, Any]:
         """Get user metadata including scopes."""
         resp = await self._request(
@@ -49,7 +63,7 @@ class Teslemetry(TeslaFleetApi):
             self.server = f"https://{self.region}.teslemetry.com"
             LOGGER.debug("Using server %s", self.server)
         return resp
-    
+
     async def scopes(self) -> list[str]:
         """Get user scopes."""
         resp = await self.metadata(False)
@@ -61,6 +75,23 @@ class Teslemetry(TeslaFleetApi):
         assert self.region
         return self.region
 
+    async def server_side_polling(self, vin: str, value: bool | None = None) -> bool | None:
+        """Get or set Auto mode."""
+        if value is True:
+            return (await self._request(
+                Method.POST,
+                f"api/auto/{vin}",
+            )).get("response")
+        if value is False:
+            return (await self._request(
+                Method.DELETE,
+                f"api/auto/{vin}",
+            )).get("response")
+        return (await self._request(
+            Method.GET,
+            f"api/auto/{vin}",
+        )).get("response")
+
     async def _request(
         self,
         method: Method,

{tesla_fleet_api-0.7.2.dist-info → tesla_fleet_api-0.7.4.dist-info}/METADATA

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

{tesla_fleet_api-0.7.2.dist-info → tesla_fleet_api-0.7.4.dist-info}/RECORD

@@ -1,20 +1,21 @@
 tesla_fleet_api/__init__.py,sha256=qGS4Qfp0uHaeK8UXOQVYjZ9rs_iR-CJ6rfCMLswphKA,667
 tesla_fleet_api/charging.py,sha256=N_mc8axrXj3iduqLj_jCt4Vx86tHqe3xqQT4R1R7HvU,1689
-tesla_fleet_api/const.py,sha256=oMxY97yVTIRA94E2AejYiZ-Xe8WCEB9NPXaT2ufjXLc,9277
+tesla_fleet_api/const.py,sha256=ASJgrHCP2DWy3GL9RgQ_ZZ5fVrkF12WTcyIHlpOxdxI,9277
 tesla_fleet_api/energy.py,sha256=yOKNPyXWKMT7Z9a5RqdNksHokLRo657SWjly1bP2_Qs,5827
 tesla_fleet_api/energyspecific.py,sha256=zggN-q0tf6EH_57bM4EuQ-IdcemKP0o-xv__LArRNnA,4147
 tesla_fleet_api/exceptions.py,sha256=0-qeJUfyUGUcm2R3_W4OuGImgshY92ApSixl_CRM8ak,11171
 tesla_fleet_api/partner.py,sha256=1vIBUaxKLIfqcC0X6VXZN0dMAzj_CLNPUMjA6QVqZ1k,1223
-tesla_fleet_api/teslafleetapi.py,sha256=EJyfUq5X-rx7ruR6hKlSMLXLERLj2SBxT-YemiCIHcs,5490
+tesla_fleet_api/ratecalculator.py,sha256=4lz8yruUeouHXh_3ezsXX-CTpIegp1T1J4VuRV_qdHA,1791
+tesla_fleet_api/teslafleetapi.py,sha256=U6kRKS0yYztQuXPcslhZxH15-G1mPMXoFKEcwKtuVY4,5496
 tesla_fleet_api/teslafleetoauth.py,sha256=OY9yRQuokYo3ts0C8Qb6Z-o9NNAGHbX9F5mHfAh50fo,4121
 tesla_fleet_api/teslafleetopensource.py,sha256=TJfVPcqJlA1b3kMoGuLr-g5Gn8UDyYsTZhjvGY1MtIk,2007
-tesla_fleet_api/teslemetry.py,sha256=kcZG7O9tsBt0BoUyCUSU9j9yCbN1qJDL1iUUo4DFESs,2167
+tesla_fleet_api/teslemetry.py,sha256=tXGfZmF0lO3fdgdoR2T8mXlWuF4dPKZebGijkwQN5Mk,3138
 tesla_fleet_api/tessie.py,sha256=4dBYxe1G2v9JvJGRbb01wXrAmvWT4jOfV4f_VQE_vkE,2302
 tesla_fleet_api/user.py,sha256=TZE2oh-n5zrhKXmGRuiNL9voKVODD7rBhGE_IObYVGA,1179
 tesla_fleet_api/vehicle.py,sha256=KFFotHSmzaC4MhIlU8hoG7SVvPiV3_FC__uJf8BG_G0,31412
 tesla_fleet_api/vehiclespecific.py,sha256=Nr4zZzfmIuw3RFYjQEX6c_xtYZgztMsN5ohVn-YEH0I,20600
-tesla_fleet_api-0.7.2.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-tesla_fleet_api-0.7.2.dist-info/METADATA,sha256=yBS2vyPI8A7jgI2XNqDUsDangdsuAB60-0ZCO5KhzZE,3821
-tesla_fleet_api-0.7.2.dist-info/WHEEL,sha256=Z4pYXqR_rTB7OWNDYFOm1qRk0RX6GFP2o8LgvP453Hk,91
-tesla_fleet_api-0.7.2.dist-info/top_level.txt,sha256=jeNbog_1saXBFrGpom9WyPWmilxsyP3szL_G7JLWQfM,16
-tesla_fleet_api-0.7.2.dist-info/RECORD,,
+tesla_fleet_api-0.7.4.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+tesla_fleet_api-0.7.4.dist-info/METADATA,sha256=86anwGAk_8eZvWjidXnuQaWR6Kv293SE0vBL2G9LY3k,3821
+tesla_fleet_api-0.7.4.dist-info/WHEEL,sha256=UvcQYKBHoFqaQd6LKyqHw9fxEolWLQnlzP0h_LgJAfI,91
+tesla_fleet_api-0.7.4.dist-info/top_level.txt,sha256=jeNbog_1saXBFrGpom9WyPWmilxsyP3szL_G7JLWQfM,16
+tesla_fleet_api-0.7.4.dist-info/RECORD,,

{tesla_fleet_api-0.7.2.dist-info → tesla_fleet_api-0.7.4.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (70.3.0)
+Generator: setuptools (74.0.0)
 Root-Is-Purelib: true
 Tag: py3-none-any