Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 12:59:34 +00:00
Code Activity

Add index for area/config_entry/label to the device registry (#114776)

Browse Source 
* Add index for area/config_entry/label to the device registry

* use it for services

* naming

* naming

* tweak
This commit is contained in:
J. Nick Koston
2024-04-03 16:52:17 -10:00
committed by GitHub
parent 841d3940d1
commit 3f76d1f056
3 changed files with 76 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
homeassistant/helpers/service.py
View File

@@ -534,15 +534,14 @@ def async_extract_referenced_entity_ids( # noqa: C901
):
selected.indirectly_referenced.add(entity_entry.entity_id)
# Find areas, devices & entities for targeted labels
for device_entry in dev_reg.devices.get_devices_for_label(label_id):
selected.referenced_devices.add(device_entry.id)
# Find areas for targeted labels
for area_entry in area_reg.areas.values():
if area_entry.labels.intersection(selector.label_ids):
selected.referenced_areas.add(area_entry.id)
for device_entry in dev_reg.devices.values():
if device_entry.labels.intersection(selector.label_ids):
selected.referenced_devices.add(device_entry.id)
# Find areas for targeted floors
if selector.floor_ids:
for area_entry in area_reg.areas.values():
@@ -554,9 +553,11 @@ def async_extract_referenced_entity_ids( # noqa: C901
selected.referenced_areas.update(selector.area_ids)
if selected.referenced_areas:
for device_entry in dev_reg.devices.values():
if device_entry.area_id in selected.referenced_areas:
selected.referenced_devices.add(device_entry.id)
for area_id in selected.referenced_areas:
selected.referenced_devices.update(
device_entry.id
for device_entry in dev_reg.devices.get_devices_for_area_id(area_id)
)
if not selected.referenced_areas and not selected.referenced_devices:
return selected