Update async_set_temperature to handle HVAC mode (#287)

mill1000 · Jan 8, 2025 · 2cdd310 · 2cdd310
1 parent 0a6d18b
commit 2cdd310
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/custom_components/midea_ac/climate.py b/custom_components/midea_ac/climate.py
@@ -5,7 +5,8 @@
 
 import voluptuous as vol
 from homeassistant.components.climate import ClimateEntity
-from homeassistant.components.climate.const import (PRESET_AWAY, PRESET_BOOST,
+from homeassistant.components.climate.const import (ATTR_HVAC_MODE,
+                                                    PRESET_AWAY, PRESET_BOOST,
                                                     PRESET_ECO, PRESET_NONE,
                                                     PRESET_SLEEP,
                                                     ClimateEntityFeature,
@@ -286,6 +287,9 @@ async def async_set_temperature(self, **kwargs) -> None:
         self._device.target_temperature = round(temperature * 2) / 2
         await self._apply()
 
+        if (mode := kwargs.get(ATTR_HVAC_MODE, None)) is not None:
+            await self.async_set_hvac_mode(mode)
+
     @property
     def current_humidity(self) -> float | None:
         """Return the current humidity."""