feedc0de/homeassistant-core
1
0
Fork 0
forked from home-assistant/core
Code Pull Requests Activity

Fix incorrect RainMachine service helper (#58633)

Browse Source
This commit is contained in:
Aaron Bach
2021-10-28 14:29:25 -06:00
committed by Paulus Schoutsen
parent ffe008cbef
commit da2addf15a
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
homeassistant/components/rainmachine/__init__.py
View File

@@ -135,9 +135,11 @@ def async_get_controller_for_service_call(
device_registry = dr.async_get(hass) device_registry = dr.async_get(hass)
if device_entry := device_registry.async_get(device_id): if device_entry := device_registry.async_get(device_id):
for entry_id in device_entry.config_entries: for entry in hass.config_entries.async_entries(DOMAIN):
if controller := hass.data[DOMAIN][entry_id][DATA_CONTROLLER]: if entry.entry_id in device_entry.config_entries:
return cast(Controller, controller) return cast(
Controller, hass.data[DOMAIN][entry.entry_id][DATA_CONTROLLER]
)
raise ValueError(f"No controller for device ID: {device_id}") raise ValueError(f"No controller for device ID: {device_id}")