Make sure _scan_thermostats() is executed when required

Author: bouwew

plugwise/helper.py

@@ -794,9 +794,12 @@ def _cleanup_data(self, data: GwEntityData) -> None:
     def _scan_thermostats(self) -> None:
         """Helper-function for smile.py: get_all_entities().
 
-        Update locations with thermostat ranking results and use
+        Adam only: update locations with thermostat ranking results and use
         the result to update the device_class of secondary thermostats.
         """
+        if not self.check_name(ADAM):
+            return
+
         self._match_and_rank_thermostats()
         for location_id, location in self._loc_data.items():
             if location["primary_prio"] != 0:
@@ -852,7 +855,11 @@ def _rank_thermostat(
         There can be several primary and secondary thermostats per location.
         """
         appl_class = entity["dev_class"]
-        if location_id == entity["location"] and appl_class in thermo_matching:
+        if (
+            "location" in entity
+            and location_id == entity["location"]
+            and appl_class in thermo_matching
+        ):
             # Pre-elect new primary
             if thermo_matching[appl_class] == location["primary_prio"]:
                 location["primary"].append(entity_id)

plugwise/smile.py

@@ -114,9 +114,9 @@ def get_all_gateway_entities(self) -> None:
             if self._is_thermostat:
                 self.therms_with_offset_func = (
                     self._get_appliances_with_offset_functionality()
-                )
-                if self.check_name(ADAM):
-                    self._scan_thermostats()
+            )
+
+        self._scan_thermostats()
 
         self._get_groups()