home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-08-16 11:01:39 +02:00
Code Issues Packages Projects Releases Wiki Activity

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
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.
"""
ent_reg = async_get(hass)
for entry in list(ent_reg.entities.values()):
if entry.config_entry_id != config_entry_id:
continue
if not ent_reg.entities.get_entry(entry.id):
continue
updates = entry_callback(entry)
if updates is not None:
entities = ent_reg.entities
for entry in entities.get_entries_for_config_entry_id(config_entry_id):
if (
entities.get_entry(entry.id)
and (updates := entry_callback(entry)) is not None
):
ent_reg.async_update_entity(entry.entity_id, **updates)