Skip to content

Commit 7967911

Browse files
committed
Add refresh_hook to Teslemetry
1 parent 8672dbc commit 7967911

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

tesla_fleet_api/teslemetry/teslemetry.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,21 +1,22 @@
11
from typing import Any
22

33
import aiohttp
4+
from typing_extensions import Awaitable, Callable
45

5-
from tesla_fleet_api.teslemetry.vehicles import TeslemetryVehicles
66
from tesla_fleet_api.const import LOGGER, Method
77
from tesla_fleet_api.tesla import TeslaFleetApi
8-
9-
class Teslemetry(TeslaFleetApi):
8+
from tesla_fleet_api.teslemetry.vehicles import TeslemetryVehicles
109

1110

11+
class Teslemetry(TeslaFleetApi):
1212
Vehicles = TeslemetryVehicles
1313

1414
def __init__(
1515
self,
1616
session: aiohttp.ClientSession,
1717
access_token: str,
18-
server: str = "https://api.teslemetry.com"
18+
server: str = "https://api.teslemetry.com",
19+
refresh_hook: Callable[[], Awaitable[str | None]] | None = None,
1920
):
2021
"""Initialize the Teslemetry API."""
2122

@@ -28,6 +29,8 @@ def __init__(
2829
self.user = self.User(self)
2930
self.vehicles = self.Vehicles(self)
3031

32+
self.refresh_hook = refresh_hook
33+
3134
async def ping(self) -> dict[str, bool]:
3235
"""Send a ping."""
3336
return await self._request(

0 commit comments

Comments
 (0)