jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-19 19:27:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Avoid linear search of the entity registry in ps4 (#109723)

Browse Source
This commit is contained in:
J. Nick Koston 2024-02-05 16:25:12 -06:00 committed by GitHub
parent 440212ddce
commit 6fce8a5403
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 33 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
homeassistant/components/ps4/__init__.py
View File

@ -120,9 +120,11 @@ async def async_migrate_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
# Prevent changing entity_id. Updates entity registry. # Prevent changing entity_id. Updates entity registry.
registry = er.async_get(hass) registry = er.async_get(hass)
for entity_id, e_entry in registry.entities.items(): for e_entry in registry.entities.get_entries_for_config_entry_id(
if e_entry.config_entry_id == entry.entry_id: entry.entry_id
):
unique_id = e_entry.unique_id unique_id = e_entry.unique_id
entity_id = e_entry.entity_id
# Remove old entity entry. # Remove old entity entry.
registry.async_remove(entity_id) registry.async_remove(entity_id)

8
homeassistant/components/ps4/media_player.py
View File

@ -344,10 +344,12 @@ class PS4Device(MediaPlayerEntity):
_LOGGER.info("Assuming status from registry") _LOGGER.info("Assuming status from registry")
e_registry = er.async_get(self.hass) e_registry = er.async_get(self.hass)
d_registry = dr.async_get(self.hass) d_registry = dr.async_get(self.hass)
for entity_id, entry in e_registry.entities.items():
if entry.config_entry_id == self._entry_id: for entry in e_registry.entities.get_entries_for_config_entry_id(
self._entry_id
):
self._attr_unique_id = entry.unique_id self._attr_unique_id = entry.unique_id
self.entity_id = entity_id self.entity_id = entry.entity_id
break break
for device in d_registry.devices.values(): for device in d_registry.devices.values():
if self._entry_id in device.config_entries: if self._entry_id in device.config_entries: