Avoid linear search of entity registry in huawei_lte (#109637)

This commit is contained in:
J. Nick Koston 2024-02-04 14:13:06 -06:00 committed by GitHub
parent 13a5038c17
commit 3531444e2e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -70,11 +70,10 @@ async def async_setup_entry(
track_wired_clients = router.config_entry.options.get( track_wired_clients = router.config_entry.options.get(
CONF_TRACK_WIRED_CLIENTS, DEFAULT_TRACK_WIRED_CLIENTS CONF_TRACK_WIRED_CLIENTS, DEFAULT_TRACK_WIRED_CLIENTS
) )
for entity in registry.entities.values(): for entity in registry.entities.get_entries_for_config_entry_id(
if ( config_entry.entry_id
entity.domain == DEVICE_TRACKER_DOMAIN
and entity.config_entry_id == config_entry.entry_id
): ):
if entity.domain == DEVICE_TRACKER_DOMAIN:
mac = entity.unique_id.partition("-")[2] mac = entity.unique_id.partition("-")[2]
# Do not add known wired clients if not tracking them (any more) # Do not add known wired clients if not tracking them (any more)
skip = False skip = False