Fixed listeners unload after integration is removed

tomaae · Jan 24, 2022 · f1268ad · f1268ad
1 parent 556d550
commit f1268ad
Showing 1 changed file with 12 additions and 10 deletions.
diff --git a/custom_components/mikrotik_router/mikrotik_controller.py b/custom_components/mikrotik_router/mikrotik_controller.py
@@ -167,22 +167,24 @@ def __init__(self, hass, config_entry):
 
         self.major_fw_version = 0
 
-        self._force_update_callback = None
-        self._force_fwupdate_check_callback = None
-        self._async_ping_tracked_hosts_callback = None
-
         self.async_mac_lookup = AsyncMacLookup()
         # self.async_mac_lookup.update_vendors()
 
     async def async_init(self):
-        self._force_update_callback = async_track_time_interval(
-            self.hass, self.force_update, self.option_scan_interval
+        self.listeners.append(
+            async_track_time_interval(
+                self.hass, self.force_update, self.option_scan_interval
+            )
         )
-        self._force_fwupdate_check_callback = async_track_time_interval(
-            self.hass, self.force_fwupdate_check, timedelta(hours=1)
+        self.listeners.append(
+            async_track_time_interval(
+                self.hass, self.force_fwupdate_check, timedelta(hours=1)
+            )
         )
-        self._async_ping_tracked_hosts_callback = async_track_time_interval(
-            self.hass, self.async_ping_tracked_hosts, timedelta(seconds=15)
+        self.listeners.append(
+            async_track_time_interval(
+                self.hass, self.async_ping_tracked_hosts, timedelta(seconds=15)
+            )
         )
 
     # ---------------------------