refactor: Do rediscovery only during TimeoutError

davidrapan · Jan 10, 2025 · 97783df · 97783df
1 parent 036a5df
commit 97783df
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/custom_components/solarman/api.py b/custom_components/solarman/api.py
@@ -107,7 +107,7 @@ async def try_read_write(self, code, start, arg, message: str):
             except Exception as e:
                 _LOGGER.debug(f"[{self.config.serial}] {message} failed, attempts left: {attempts_left}{'' if attempts_left > 0 else ', aborting.'} [{format_exception(e)}]")
 
-                if attempts_left == ACTION_ATTEMPTS - 2:
+                if isinstance(e, TimeoutError):
                     await self.endpoint.discover(True)
                 if not self.modbus.auto_reconnect:
                     await self.modbus.disconnect()