Move another helper-function

Author: bouwew

plugwise/helper.py

@@ -25,6 +25,7 @@
     ATTR_UNIT_OF_MEASUREMENT,
     COOLING_ICON,
     DEVICE_MEASUREMENTS,
+    DEVICE_STATE,
     DOMAIN_OBJECTS,
     ENERGY_KILO_WATT_HOUR,
     ENERGY_WATT_HOUR,
@@ -35,6 +36,7 @@
     HOME_MEASUREMENTS,
     LOCATIONS,
     POWER_WATT,
+    PW_NOTIFICATION,
     SWITCH_GROUP_TYPES,
     THERMOSTAT_CLASSES,
 )
@@ -1081,3 +1083,13 @@ def pw_notification_updater(self, d_id, d_dict):
                 self.gw_devices[d_id]["binary_sensors"][idx][ATTR_STATE] = (
                     self.notifications != {}
                 )
+
+    def append_special(self, data, d_id, bs_list, s_list):
+        """Helper for all_device_data()."""
+        if d_id == self.gateway_id:
+            if self.single_master_thermostat() is not None:
+                bs_list.append(PW_NOTIFICATION)
+            if not self.active_device_present and "heating_state" in data:
+                s_list.append(DEVICE_STATE)
+        if d_id == self.heater_id and self.single_master_thermostat() is False:
+            s_list.append(DEVICE_STATE)

plugwise/smile.py

@@ -234,16 +234,6 @@ async def update_gw_devices(self):
                 for key, value in list(data.items()):
                     self.update_helper(data, dev_dict, dev_id, "switches", key)
 
-    def append_special(self, data, dev_id, bs_list, s_list):
-        """Helper for all_device_data()."""
-        if dev_id == self.gateway_id:
-            if self.single_master_thermostat() is not None:
-                bs_list.append(PW_NOTIFICATION)
-            if not self.active_device_present and "heating_state" in data:
-                s_list.append(DEVICE_STATE)
-        if dev_id == self.heater_id and self.single_master_thermostat() is False:
-            s_list.append(DEVICE_STATE)
-
     def all_device_data(self):
         "Helper-function: collect data for each device and add to self.gw_devices."
         dev_id_list = []