Split off _async_remove_devices() function

Author: bouwew

custom_components/plugwise/coordinator.py

@@ -118,16 +118,18 @@ async def _async_update_data(self) -> PlugwiseData:
 
         return data
 
-    def _async_add_remove_devices(self, data:PlugwiseData, entry: ConfigEntry,) -> None:
+    def _async_add_remove_devices(self, data: PlugwiseData, entry: ConfigEntry) -> None:
         """Add new Plugwise devices, remove non-existing devices."""
         # Check for new or removed devices
         self.new_devices = set(data.devices) - self._current_devices
         removed_devices = self._current_devices - set(data.devices)
         self._current_devices = set(data.devices)
-        if not removed_devices:
-            return
 
-        # Clean device_registry when removed devices found
+        if removed_devices:
+            self._async_remove_devices(data, entry)
+
+    def _async_remove_devices(self, data: PlugwiseData, entry: ConfigEntry) -> None:
+        """Clean registries when removed devices found."""
         device_reg = dr.async_get(self.hass)
         device_list = dr.async_entries_for_config_entry(
             device_reg, self.config_entry.entry_id