Merge pull request #81 from zabuldon/dev

fix: add retry to get_vehicles
zabuldon · May 31, 2020 · aeaeb04 · aeaeb04
2 parents 62a4bd9 + 9c440c8
commit aeaeb04
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/teslajsonpy/controller.py b/teslajsonpy/controller.py
@@ -12,6 +12,7 @@
 
 import backoff
 import wrapt
+from aiohttp import ClientConnectorError
 
 from teslajsonpy.battery_sensor import Battery, Range
 from teslajsonpy.binary_sensor import (
@@ -344,6 +345,7 @@ def register_websocket_callback(self, callback) -> int:
         self.__websocket_listeners.append(callback)
         return len(self.__websocket_listeners) - 1
 
+    @backoff.on_exception(min_expo, ClientConnectorError, max_time=10, logger=__name__)
     async def get_vehicles(self):
         """Get vehicles json from TeslaAPI."""
         return (await self.__connection.get("vehicles"))["response"]