Handle None-cases

Author: bouwew

plugwise/helper.py

@@ -798,7 +798,7 @@ def _control_state(self, data: GwEntityData) -> str | bool:
         Note: heating or cooling can still be active when the setpoint has been reached.
         """
 
-        if (ctrl_state := data["thermostat"]["control_state"]) is not None:
+        if (ctrl_state := data["thermostat"].get("control_state")) is not None:
             data["thermostat"].pop("control_state")
             return ctrl_state
 
@@ -839,9 +839,10 @@ def _regulation_control(self, data: GwEntityData) -> None:
 
         Adam: collect the thermostat regulation_mode of a location.
         """
-        data["regulation_control_modes"] = ["active", "passive", "off"]
-        data["select_regulation_control"] = data["thermostat"]["regulation_control"]
-        data["thermostat"].pop("regulation_control")
+        if (reg_control := data["thermostat"].get("regulation_control")) is not None:
+            data["select_regulation_control"] = reg_control
+            data["regulation_control_modes"] = ["active", "passive", "off"]
+            data["thermostat"].pop("regulation_control")
 
     def _preset(self, loc_id: str) -> str | None:
         """Helper-function for smile.py: device_data_climate().