feat: use macro define for fan entity

XiaoMi · Jan 22, 2025 · 035edcd · 035edcd
1 parent 517542a
commit 035edcd
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/custom_components/xiaomi_home/fan.py b/custom_components/xiaomi_home/fan.py
@@ -52,7 +52,11 @@
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
-from homeassistant.components.fan import FanEntity, FanEntityFeature
+from homeassistant.components.fan import (
+    FanEntity,
+    FanEntityFeature,
+    DIRECTION_REVERSE
+)
 from homeassistant.util.percentage import (
     percentage_to_ranged_value,
     ranged_value_to_percentage,
@@ -265,7 +269,7 @@ async def async_set_direction(self, direction: str) -> None:
             prop=self._prop_wind_reverse,
             value=(
                 self._prop_wind_reverse_reverse
-                if self.current_direction == 'reverse'
+                if self.current_direction == DIRECTION_REVERSE
                 else self._prop_wind_reverse_forward))
 
     async def async_oscillate(self, oscillating: bool) -> None: