Fix for pw-beta issue #776

Author: bouwew

plugwise/helper.py

@@ -917,20 +917,19 @@ def _rank_thermostat(
             else:
                 thermo_loc["secondary"].append(appliance_id)
 
-    def _control_state(self, loc_id: str) -> str | bool:
+    def _control_state(self, loc_id: str) -> str:
         """Helper-function for _get_adam_data().
 
         Adam: find the thermostat control_state of a location, from DOMAIN_OBJECTS.
         Represents the heating/cooling demand-state of the local primary thermostat.
         Note: heating or cooling can still be active when the setpoint has been reached.
         """
-        locator = f'location[@id="{loc_id}"]'
-        if (location := self._domain_objects.find(locator)) is not None:
-            locator = './actuator_functionalities/thermostat_functionality[type="thermostat"]/control_state'
-            if (ctrl_state := location.find(locator)) is not None:
-                return str(ctrl_state.text)
+        locator = f'location[@id="{loc_id}"]/actuator_functionalities/thermostat_functionality[type="thermostat"]/control_state'
+        if (ctrl_state := self._domain_objects.find(locator)) is not None:
+            return str(ctrl_state.text)
 
-        return False
+        # When the key is not found the regulation_mode is set to off (#776)
+        return OFF
 
     def _heating_valves(self) -> int | bool:
         """Helper-function for smile.py: _get_adam_data().