jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-04-25 17:57:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

Avoid linear search of entity registry in keenetic_ndms2 (#109635)

Browse Source
This commit is contained in:
J. Nick Koston 2024-02-04 14:12:24 -06:00 committed by GitHub
parent c988d3d427
commit 6ccf82d7b1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 8 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
homeassistant/components/keenetic_ndms2/__init__.py
View File

@ -75,11 +75,10 @@ async def async_unload_entry(hass: HomeAssistant, config_entry: ConfigEntry) ->
for mac, device in router.last_devices.items()
if device.interface in new_tracked_interfaces
}
for entity_entry in list(ent_reg.entities.values()):
if (
entity_entry.config_entry_id == config_entry.entry_id
and entity_entry.domain == Platform.DEVICE_TRACKER
):
for entity_entry in ent_reg.entities.get_entries_for_config_entry_id(
config_entry.entry_id
):
if entity_entry.domain == Platform.DEVICE_TRACKER:
mac = entity_entry.unique_id.partition("_")[0]
if mac not in keep_devices:
_LOGGER.debug("Removing entity %s", entity_entry.entity_id)

9
homeassistant/components/keenetic_ndms2/device_tracker.py
View File

@ -43,11 +43,10 @@ async def async_setup_entry(
registry = er.async_get(hass)
# Restore devices that are not a part of active clients list.
restored = []
for entity_entry in registry.entities.values():
if (
entity_entry.config_entry_id == config_entry.entry_id
and entity_entry.domain == DEVICE_TRACKER_DOMAIN
):
for entity_entry in registry.entities.get_entries_for_config_entry_id(
config_entry.entry_id
):
if entity_entry.domain == DEVICE_TRACKER_DOMAIN:
mac = entity_entry.unique_id.partition("_")[0]
if mac not in tracked:
tracked.add(mac)