Add retry behaviour.

pulquero · Oct 18, 2024 · 3a67a47 · 3a67a47
1 parent 6aef0dc
commit 3a67a47
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/modules/teslapy_wrapper.py b/src/modules/teslapy_wrapper.py
@@ -10,7 +10,8 @@
 
 @pyscript_executor
 def set_powerwall_tariff(email, refresh_token, tariff_data):
-    with teslapy.Tesla(email) as tesla:
+    retry = teslapy.Retry(total=5, allowed_methods=None, backoff_factor=1, status_forcelist=(503, 504))
+    with teslapy.Tesla(email, retry=retry) as tesla:
         tesla.refresh_token(refresh_token=refresh_token)
         pw = tesla.battery_list()[0]
         pw.set_tariff(tariff_data)
@@ -26,7 +27,8 @@ def get_powerwall_tariff(email, refresh_token):
 
 @pyscript_executor
 def set_powerwall_settings(email, refresh_token, reserve_percentage=None, mode=None, allow_grid_charging=None, allow_battery_export=None):
-    with teslapy.Tesla(email) as tesla:
+    retry = teslapy.Retry(total=5, allowed_methods=None, backoff_factor=1, status_forcelist=(503, 504))
+    with teslapy.Tesla(email, retry=retry) as tesla:
         tesla.refresh_token(refresh_token=refresh_token)
         pw = tesla.battery_list()[0]
         if reserve_percentage is not None: