Fix/guard/add count

Author: bouwew

plugwise/data.py

@@ -144,9 +144,10 @@ def _update_for_cooling(self, entity: GwEntityData) -> None:
             entity["thermostat"] = temp_dict
             if "setpoint" in sensors:
                 sensors.pop("setpoint")
+                self._count -= 1
             sensors["setpoint_low"] = temp_dict["setpoint_low"]
             sensors["setpoint_high"] = temp_dict["setpoint_high"]
-            self._count += 2  # add 4, remove 2
+            self._count += 3  # add 4, remove 1
 
     def _get_location_data(self, loc_id: str) -> GwEntityData:
         """Helper-function for _all_entity_data() and async_update().
@@ -164,8 +165,9 @@ def _get_location_data(self, loc_id: str) -> GwEntityData:
         ):
             data["control_state"] = str(ctrl_state)
 
-        data["sensors"].pop("setpoint")  # remove, only used in _control_state()
-        self._count -= 1
+        if "setpoint" in data["sensors"]:
+            data["sensors"].pop("setpoint")  # remove, only used in _control_state()
+            self._count -= 1
 
         # Thermostat data (presets, temperatures etc)
         self._climate_data(loc_id, zone, data)
@@ -238,6 +240,7 @@ def _get_adam_data(self, entity: GwEntityData, data: GwEntityData) -> None:
                 and self._cooling_present
             ):
                 data["binary_sensors"]["cooling_enabled"] = self._cooling_enabled
+                self._count += 1
 
         # Show the allowed regulation_modes and gateway_modes
         if entity["dev_class"] == "gateway":