feat(wip): Trigger reauth after data update token expired

duhow · Mar 6, 2023 · f7e97dc · f7e97dc
1 parent 78768a0
commit f7e97dc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/custom_components/aigues_barcelona/config_flow.py b/custom_components/aigues_barcelona/config_flow.py
@@ -101,7 +101,13 @@ async def async_step_reauth(self, entry) -> FlowResult:
         """Request OAuth Token again when expired."""
         # get previous entity content back to flow
         self.entry = entry
-        self.stored_input = entry.data
+        if hasattr(entry, "data"):
+            self.stored_input = entry.data
+        else:
+            # FIXME: for DataUpdateCoordinator, entry is not valid,
+            # as it contains only sensor data. Missing entry_id.
+            # Reauth when restarting works.
+            self.stored_input = entry
         return await self.async_step_reauth_confirm(None)
 
     async def async_step_reauth_confirm(

diff --git a/custom_components/aigues_barcelona/sensor.py b/custom_components/aigues_barcelona/sensor.py
@@ -122,9 +122,9 @@ async def _async_update_data(self):
             consumptions = await self.hass.async_add_executor_job(
                 self._api.consumptions, LAST_WEEK, TOMORROW
             )
-        except ConfigEntryAuthFailed:
+        except ConfigEntryAuthFailed as exp:
             _LOGGER.error("Token has expired, cannot check consumptions.")
-            return False
+            raise ConfigEntryAuthFailed from exp
         except Exception as msg:
             _LOGGER.error("error while getting data: %s", msg)