Fix mistake in _get_toggle_state()

plugwise · Nov 30, 2023 · e3d95d1 · e3d95d1
1 parent bc4f47d
commit e3d95d1
Showing 1 changed file with 9 additions and 12 deletions.
diff --git a/plugwise/helper.py b/plugwise/helper.py
@@ -1555,15 +1555,12 @@ def _get_toggle_state(
         Obtain the toggle state of a 'toggle' = switch.
         """
         if xml.find("type").text == "heater_central":
-            locator = "./actuator_functionalities/toggle_functionality"
-            if found := xml.findall(locator):
-                for item in found:
-                    if (toggle_type := item.find("type")) is not None:
-                        if toggle_type.text == toggle:
-                            data["switches"][name] = item.find("state").text == "on"
-                            self._count += 1
-                            # Remove the cooling_enabled binary_sensor when the corresponding switch is present
-                            # Except for Elga
-                            if toggle == "cooling_enabled" and not self._elga:
-                                data["binary_sensors"].pop("cooling_enabled")
-                                self._count -= 1
+            locator = f"./actuator_functionalities/toggle_functionality[type='{toggle}']/state"
+            if (state := xml.find(locator)) is not None:
+                data["switches"][name] = state.text == "on"
+                self._count += 1
+                # Remove the cooling_enabled binary_sensor when the corresponding switch is present
+                # Except for Elga
+                if toggle == "cooling_enabled" and not self._elga:
+                    data["binary_sensors"].pop("cooling_enabled")
+                    self._count -= 1