Add: Retry 3 times on update failure before raising reauth request

JurajNyiri · Nov 19, 2023 · d62c505 · d62c505
1 parent eeee05b
commit d62c505
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/custom_components/tapo_control/__init__.py b/custom_components/tapo_control/__init__.py
@@ -449,11 +449,18 @@ async def async_update_data():
                         updateDataForAllControllers[controller] = await getCamData(
                             hass, controller
                         )
+                        hass.data[DOMAIN][entry.entry_id]["reauth_retries"] = 0
                     except Exception as e:
                         updateDataForAllControllers[controller] = False
                         if str(e) == "Invalid authentication data":
-                            hass.data[DOMAIN][entry.entry_id]["refreshEnabled"] = False
-                            raise ConfigEntryAuthFailed(e)
+                            if hass.data[DOMAIN][entry.entry_id]["reauth_retries"] < 3:
+                                hass.data[DOMAIN][entry.entry_id]["reauth_retries"] += 1
+                                raise e
+                            else:
+                                hass.data[DOMAIN][entry.entry_id][
+                                    "refreshEnabled"
+                                ] = False
+                                raise ConfigEntryAuthFailed(e)
                         LOGGER.error(e)
 
                 hass.data[DOMAIN][entry.entry_id][
@@ -552,6 +559,7 @@ async def async_update_data():
 
         hass.data[DOMAIN][entry.entry_id] = {
             "setup_retries": 0,
+            "reauth_retries": 0,
             "runningMediaSync": False,
             "controller": tapoController,
             "entry": entry,