turn fan_speed to property

IATkachenko · IATkachenko · commit 70ee1236de18 · 2020-06-16T14:14:11.000+03:00
It will allow use constructions like
tion = tion.s3(mac)
tion.fan_speed = 4
i.e. don't use set/get functions directly
diff --git a/tion/s3.py b/tion/s3.py
@@ -65,9 +65,10 @@ def process_mode(mode_code: int) -> str:
 
             result = {}
             try:
+                self.fan_speed = int(list("{:02x}".format(response[2]))[1])
                 result = {"code": 200,
                           "mode": process_mode(int(list("{:02x}".format(response[2]))[0])),
-                          "fan_speed": int(list("{:02x}".format(response[2]))[1]),
+                          "fan_speed": self.fan_speed,
                           "heater_temp": response[3],
                           "heater": self._process_status(response[4] & 1),
                           "status": self._process_status(response[4] >> 1 & 1),
diff --git a/tion/tion.py b/tion/tion.py
@@ -43,6 +43,7 @@ def __init__(self, mac: str):
         self._mac = mac
         self._btle: btle.Peripheral = btle.Peripheral(None)
         self._delegation = TionDelegation()
+        self._fan_speed = 0
 
     @abc.abstractmethod
     def _send_request(self, request: bytearray) -> bytearray:
@@ -184,3 +185,18 @@ def _enable_notifications(self):
         self._btle.withDelegate(self._delegation)
         self.notify.read()
         return result
+
+    @property
+    def fan_speed(self):
+        return self._fan_speed
+
+    @fan_speed.setter
+    def fan_speed(self, new_speed: int):
+        if 0 <= new_speed <= 6:
+            self._fan_speed = new_speed
+
+        else:
+            _LOGGER.warning("Incorrect new fan speed. Will use 1 instead")
+            self._fan_speed = 1
+
+        # self.set({"fan_speed": new_speed})
