mts_working : moving it to the inherited (MTS960) thermostat class in…

…stead of the base MTSClimate
krahabb · Jun 9, 2024 · 5682fb5 · 5682fb5
1 parent 8ced678
commit 5682fb5
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 11 deletions.
diff --git a/custom_components/meross_lan/climate.py b/custom_components/meross_lan/climate.py
@@ -96,7 +96,6 @@ class MtsClimate(me.MerossEntity, climate.ClimateEntity):
         "_mts_mode",
         "_mts_onoff",
         "_mts_payload",
-        "_mts_working",
         "_mts_adjust_offset",
         "number_adjust_temperature",
         "number_preset_temperature",
@@ -122,7 +121,6 @@ def __init__(
         self._mts_active = None
         self._mts_mode: int | None = None
         self._mts_onoff: int | None = None
-        self._mts_working: int | None = None
         self._mts_payload = {}
         self._mts_adjust_offset = 0
         super().__init__(manager, channel)
@@ -148,7 +146,6 @@ async def async_shutdown(self):
     def set_unavailable(self):
         self._mts_active = None
         self._mts_mode = None
-        self._mts_working = None
         self._mts_onoff = None
         self._mts_payload = {}
         self.preset_mode = None

diff --git a/custom_components/meross_lan/devices/mts960.py b/custom_components/meross_lan/devices/mts960.py
@@ -111,7 +111,7 @@ class Mts960Climate(MtsClimate):
 
     preset_modes = list(MTS_MODE_TO_PRESET_MAP.values())
 
-    __slots__ = ()
+    __slots__ = ( "_mts_working" )
 
     def __init__(
         self,
@@ -126,6 +126,12 @@ def __init__(
             None,
             Mts960Schedule,
         )
+        self._mts_working = None
+
+    def set_unavailable(self):
+        self._mts_working = None
+        super().set_unavailable()
+
 
     def flush_state(self):
         """interface: MtsClimate."""

diff --git a/tests/entities/climate.py b/tests/entities/climate.py
@@ -31,13 +31,13 @@
         mc.MTS200_MODE_MANUAL,
     },
     Mts960Climate: {
-        mc.MTS960_MODE_HEAT,
-        mc.MTS960_MODE_COOL,
-        mc.MTS960_MODE_CYCLE,
-        mc.MTS960_MODE_COUNTDOWN_ON,
-        mc.MTS960_MODE_COUNTDOWN_OFF,
-        mc.MTS960_MODE_SCHEDULE_HEAT,
-        mc.MTS960_MODE_SCHEDULE_COOL,
+        # mc.MTS960_MODE_HEAT,
+        # mc.MTS960_MODE_COOL,
+        # mc.MTS960_MODE_CYCLE,
+        # mc.MTS960_MODE_COUNTDOWN_ON,
+        # mc.MTS960_MODE_COUNTDOWN_OFF,
+        # mc.MTS960_MODE_SCHEDULE_HEAT,
+        # mc.MTS960_MODE_SCHEDULE_COOL,
     },
 }