Guard for missing thermostat data

Author: bouwew

plugwise/helper.py

@@ -820,8 +820,11 @@ def _control_state(self, data: GwEntityData) -> str | bool:
         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.
         """
+        if (thermostat := data.get("thermostat")) is None:
+            LOGGER.warning("Something wrong, incomplete thermostat data")
+            return
 
-        if (ctrl_state := data["thermostat"].get("control_state")) is not None:
+        if (ctrl_state := thermostat.get("control_state")) is not None:
             data["thermostat"].pop("control_state")
             self._count -= 1
             return ctrl_state
@@ -863,8 +866,11 @@ def _regulation_control(self, data: GwEntityData) -> None:
 
         Adam: collect the thermostat regulation_control state of a location.
         """
-        LOGGER.debug("HOI data: %s", data)
-        if (reg_control := data["thermostat"].get("regulation_control")) is not None:
+        if (thermostat := data.get("thermostat")) is None:
+            LOGGER.warning("Something wrong, incomplete thermostat data")
+            return
+
+        if (reg_control := thermostat.get("regulation_control")) is not None:
             data["select_zone_profile"] = reg_control
             data["zone_profiles"] = ALLOWED_ZONE_PROFILES
             data["thermostat"].pop("regulation_control")