Branch p9.

Author: janiversen

custom_components/modbus/entity.py

@@ -90,7 +90,6 @@ def __init__(
         self._address = int(entry[CONF_ADDRESS])
         self._input_type = entry[CONF_INPUT_TYPE]
         self._scan_interval = int(entry[CONF_SCAN_INTERVAL])
-        self._cancel_timer: Callable[[], None] | None = None
         self._cancel_call: Callable[[], None] | None = None
 
         self._attr_unique_id = entry.get(CONF_UNIQUE_ID)
@@ -124,6 +123,21 @@ async def async_update(self, now: datetime | None = None) -> None:
                 self.hass, timedelta(seconds=self._scan_interval), self.async_update
             )
 
+    async def async_will_remove_from_hass(self) -> None:
+        """Remove entity from hass."""
+        _LOGGER.info(f"Removing entity {self._attr_name}")
+        if self._cancel_call:
+            self._cancel_call()
+            self._cancel_call = None
+
+    @callback
+    def async_hold(self) -> None:
+        """Remote stop entity."""
+        _LOGGER.info(f"hold entity {self._attr_name}")
+        self._async_cancel_future_pending_update()
+        self._attr_available = False
+        self.async_write_ha_state()
+
     async def _async_update_write_state(self) -> None:
         """Update the entity state and write it to the state machine."""
         if self._cancel_call:
@@ -143,7 +157,7 @@ async def _async_update_if_not_in_progress(
     @callback
     def async_run(self) -> None:
         """Remote start entity."""
-        self._async_cancel_update_polling()
+        _LOGGER.info(f"start entity {self._attr_name}")
         self._async_schedule_future_update(0.1)
         self._cancel_call = async_call_later(
             self.hass, timedelta(seconds=0.1), self.async_update
@@ -166,20 +180,6 @@ def _async_cancel_future_pending_update(self) -> None:
             self._cancel_call()
             self._cancel_call = None
 
-    def _async_cancel_update_polling(self) -> None:
-        """Cancel the polling."""
-        if self._cancel_timer:
-            self._cancel_timer()
-            self._cancel_timer = None
-
-    @callback
-    def async_hold(self) -> None:
-        """Remote stop entity."""
-        self._async_cancel_future_pending_update()
-        self._async_cancel_update_polling()
-        self._attr_available = False
-        self.async_write_ha_state()
-
     async def async_await_connection(self, _now: Any) -> None:
         """Wait for first connect."""
         await self._hub.event_connected.wait()

custom_components/modbus/manifest.json

@@ -6,5 +6,5 @@
   "iot_class": "local_polling",
   "loggers": ["pymodbus"],
   "requirements": ["pymodbus==3.9.2"],
-  "version": "1.8.0"
+  "version": "1.9.0"
 }