starred/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2026-09-25 07:51:46 -05:00
Code Issues Packages Projects Releases Wiki Activity

Use get_device_and_config_entry_for_domain in rainmachine (#180107)

Browse Source
This commit is contained in:
Erik Montnemery
2026-08-25 10:29:11 +02:00
committed by GitHub
parent 62a6c1f872
commit cbc7cf23ee
1 changed files with 6 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
homeassistant/components/rainmachine/services.py
+6 -8
View File
@@ -132,17 +132,15 @@ def async_get_entry_for_service_call(
) -> RainMachineConfigEntry:
"""Get the controller related to a service call (by device ID)."""
device_id = call.data[CONF_DEVICE_ID]
device_registry = dr.async_get(hass)
if (device_entry := device_registry.async_get(device_id)) is None:
device, config_entry = dr.async_get_device_and_config_entry_for_domain(
hass, device_id, domain=DOMAIN
)
if device is None:
raise ValueError(f"Invalid RainMachine device ID: {device_id}")
entry: RainMachineConfigEntry | None
for entry_id in device_entry.config_entries:
if (entry := hass.config_entries.async_get_entry(entry_id)) is None:
continue
if entry.domain == DOMAIN and entry.state is ConfigEntryState.LOADED:
return entry
if config_entry is not None and config_entry.state is ConfigEntryState.LOADED:
return config_entry
raise ValueError(f"No controller for device ID: {device_id}")