feedc0de/homeassistant-core
1
0
Fork 0
forked from home-assistant/core
Code Pull Requests Activity

Small code optimization for Plugwise (#145037)

Browse Source
This commit is contained in:
Bouwe Westerdijk
2025-05-16 13:28:31 +02:00
committed by GitHub
parent 2ca9d4689e
commit 38cee53999
1 changed files with 6 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
homeassistant/components/plugwise/coordinator.py
View File

@@ -99,12 +99,10 @@ class PlugwiseDataUpdateCoordinator(DataUpdateCoordinator[dict[str, GwEntityData
translation_key="unsupported_firmware",
) from err
self._async_add_remove_devices(data, self.config_entry)
self._async_add_remove_devices(data)
return data
def _async_add_remove_devices(
self, data: dict[str, GwEntityData], entry: ConfigEntry
) -> None:
def _async_add_remove_devices(self, data: dict[str, GwEntityData]) -> None:
"""Add new Plugwise devices, remove non-existing devices."""
# Check for new or removed devices
self.new_devices = set(data) - self._current_devices
@@ -112,11 +110,9 @@ class PlugwiseDataUpdateCoordinator(DataUpdateCoordinator[dict[str, GwEntityData
self._current_devices = set(data)
if removed_devices:
self._async_remove_devices(data, entry)
self._async_remove_devices(data)
def _async_remove_devices(
self, data: dict[str, GwEntityData], entry: ConfigEntry
) -> None:
def _async_remove_devices(self, data: dict[str, GwEntityData]) -> None:
"""Clean registries when removed devices found."""
device_reg = dr.async_get(self.hass)
device_list = dr.async_entries_for_config_entry(
@@ -136,7 +132,8 @@ class PlugwiseDataUpdateCoordinator(DataUpdateCoordinator[dict[str, GwEntityData
and identifier[1] not in data
):
device_reg.async_update_device(
device_entry.id, remove_config_entry_id=entry.entry_id
device_entry.id,
remove_config_entry_id=self.config_entry.entry_id,
)
LOGGER.debug(
"Removed %s device %s %s from device_registry",