fix naming fan modes

svmironov · Jul 12, 2023 · 7500f0f · 7500f0f
1 parent 9ecab92
commit 7500f0f
Showing 1 changed file with 13 additions and 19 deletions.
diff --git a/custom_components/aircloud/climate.py b/custom_components/aircloud/climate.py
@@ -2,6 +2,7 @@
 from homeassistant.components.climate import ClimateEntity
 from homeassistant.components.climate.const import (FAN_AUTO, FAN_HIGH,
                                                     FAN_LOW, FAN_MEDIUM,
+                                                    FAN_MIDDLE,
                                                     HVAC_MODE_AUTO,
                                                     HVAC_MODE_COOL,
                                                     HVAC_MODE_DRY,
@@ -19,11 +20,10 @@
 
 SUPPORT_FAN = [
     FAN_AUTO,
-    '1',
-    '2',
-    '3',
-    '4',
-    '5'
+    FAN_LOW,
+    FAN_MEDIUM,
+    FAN_MIDDLE,
+    FAN_HIGH
 ]
 SUPPORT_SWING = [
     SWING_OFF,
@@ -127,15 +127,13 @@ def fan_mode(self):
         if self._fan_speed == "AUTO":
             return FAN_AUTO
         elif self._fan_speed == "LV1":
-            return '1'
+            return FAN_LOW
         elif self._fan_speed == "LV2":
-            return '2'
+            return FAN_MEDIUM
         elif self._fan_speed == "LV3":
-            return '3'
+            return FAN_MIDDLE
         elif self._fan_speed == "LV4":
-            return '4'
-        elif self._fan_speed == "LV5":
-            return '5'
+            return FAN_HIGH
         else:
             return FAN_AUTO
 
@@ -192,18 +190,14 @@ def set_fan_mode(self, fan_mode):
 
         if fan_mode == FAN_AUTO:
             self._fan_speed = "AUTO"
-        elif fan_mode == 'AUTO':
-            self._fan_speed = "AUTO"
-        elif fan_mode == '1':
+        elif fan_mode == FAN_LOW:
             self._fan_speed = "LV1"
-        elif fan_mode == '2':
+        elif fan_mode == FAN_MIDDLE:
             self._fan_speed = "LV2"
-        elif fan_mode == '3':
+        elif fan_mode == FAN_MEDIUM:
             self._fan_speed = "LV3"
-        elif fan_mode == '4':
+        elif fan_mode == FAN_HIGH:
             self._fan_speed = "LV4"
-        elif fan_mode == '5':
-            self._fan_speed = "LV5"
         else:
             self._fan_speed = "AUTO"