evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use get_entries_for_config_entry_id helper in async_migrate_entries (#109629)

Browse source
This commit is contained in:
J. Nick Koston 2024-02-04 14:05:48 -06:00 • committed by GitHub
parent 7e3001f843
commit b24b4fc237
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

16
homeassistant/helpers/entity_registry.py
View file

@ -1379,16 +1379,12 @@ async def async_migrate_entries(
Can also be used to remove duplicated entity registry entries. Can also be used to remove duplicated entity registry entries.
""" """
ent_reg = async_get(hass) ent_reg = async_get(hass)
entities = ent_reg.entities
for entry in list(ent_reg.entities.values()): for entry in entities.get_entries_for_config_entry_id(config_entry_id):
if entry.config_entry_id != config_entry_id: if (
continue entities.get_entry(entry.id)
if not ent_reg.entities.get_entry(entry.id): and (updates := entry_callback(entry)) is not None
continue ):
updates = entry_callback(entry)
if updates is not None:
ent_reg.async_update_entity(entry.entity_id, **updates) ent_reg.async_update_entity(entry.entity_id, **updates)