From da2addf15a74884c43b34870ff745e5025073d75 Mon Sep 17 00:00:00 2001 From: Aaron Bach Date: Thu, 28 Oct 2021 14:29:25 -0600 Subject: [PATCH] Fix incorrect RainMachine service helper (#58633) --- homeassistant/components/rainmachine/__init__.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/rainmachine/__init__.py b/homeassistant/components/rainmachine/__init__.py index db9177591ca..b72fe0fb25d 100644 --- a/homeassistant/components/rainmachine/__init__.py +++ b/homeassistant/components/rainmachine/__init__.py @@ -135,9 +135,11 @@ def async_get_controller_for_service_call( device_registry = dr.async_get(hass) if device_entry := device_registry.async_get(device_id): - for entry_id in device_entry.config_entries: - if controller := hass.data[DOMAIN][entry_id][DATA_CONTROLLER]: - return cast(Controller, controller) + for entry in hass.config_entries.async_entries(DOMAIN): + if entry.entry_id in device_entry.config_entries: + return cast( + Controller, hass.data[DOMAIN][entry.entry_id][DATA_CONTROLLER] + ) raise ValueError(f"No controller for device ID: {device_id}")