refactor: Enhance coordinator

davidrapan · Feb 10, 2025 · d9236cb · d9236cb
1 parent 2018683
commit d9236cb
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/custom_components/solarman/coordinator.py b/custom_components/solarman/coordinator.py
@@ -17,7 +17,6 @@ def __init__(self, hass: HomeAssistant, device: Device):
         super().__init__(hass, _LOGGER, name = device.config.name, update_interval = TIMINGS_UPDATE_INTERVAL, always_update = False)
         self.device = device
         self._counter = 0
-        self._last_successful_data: dict[str, Any] | None = None
         self._is_offline_logged = False
 
     async def _async_setup(self) -> None:
@@ -32,7 +31,6 @@ async def _async_update_data(self) -> dict[str, Any]:
         try:
             try:
                 data = await self.device.get(int(self._counter * self._update_interval_seconds))
-                self._last_successful_data = data
                 self._is_offline_logged = False
                 return data
                 #return await self.device.get(int(self._counter * self._update_interval_seconds))
@@ -45,8 +43,8 @@ async def _async_update_data(self) -> dict[str, Any]:
                 _LOGGER.warning("Error retrieving data. Use last known data.")
                 self._is_offline_logged = True
 
-            if self._last_successful_data is not None:
-                return self._last_successful_data
+            if self.last_update_success:
+                return self.data
 
             if isinstance(e, TimeoutError):
                 raise