Use climate enums in tuya (#70747)

homeassistant/components/tuya/climate.py

@@ -6,23 +6,15 @@ from typing import Any
 
 from tuya_iot import TuyaDevice, TuyaDeviceManager
 
-from homeassistant.components.climate import (
-    ClimateEntity,
-    ClimateEntityDescription,
-    ClimateEntityFeature,
-)
+from homeassistant.components.climate import ClimateEntity, ClimateEntityDescription
 from homeassistant.components.climate.const import (
-    HVAC_MODE_COOL,
-    HVAC_MODE_DRY,
-    HVAC_MODE_FAN_ONLY,
-    HVAC_MODE_HEAT,
-    HVAC_MODE_HEAT_COOL,
-    HVAC_MODE_OFF,
     SWING_BOTH,
     SWING_HORIZONTAL,
     SWING_OFF,
     SWING_ON,
     SWING_VERTICAL,
+    ClimateEntityFeature,
+    HVACMode,
 )
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import TEMP_CELSIUS, TEMP_FAHRENHEIT
@@ -35,14 +27,14 @@ from .base import IntegerTypeData, TuyaEntity
 from .const import DOMAIN, LOGGER, TUYA_DISCOVERY_NEW, DPCode, DPType
 
 TUYA_HVAC_TO_HA = {
-    "auto": HVAC_MODE_HEAT_COOL,
-    "cold": HVAC_MODE_COOL,
-    "freeze": HVAC_MODE_COOL,
-    "heat": HVAC_MODE_HEAT,
-    "hot": HVAC_MODE_HEAT,
-    "manual": HVAC_MODE_HEAT_COOL,
-    "wet": HVAC_MODE_DRY,
-    "wind": HVAC_MODE_FAN_ONLY,
+    "auto": HVACMode.HEAT_COOL,
+    "cold": HVACMode.COOL,
+    "freeze": HVACMode.COOL,
+    "heat": HVACMode.HEAT,
+    "hot": HVACMode.HEAT,
+    "manual": HVACMode.HEAT_COOL,
+    "wet": HVACMode.DRY,
+    "wind": HVACMode.FAN_ONLY,
 }
 
 
@@ -50,7 +42,7 @@ TUYA_HVAC_TO_HA = {
 class TuyaClimateSensorDescriptionMixin:
     """Define an entity description mixin for climate entities."""
 
-    switch_only_hvac_mode: str
+    switch_only_hvac_mode: HVACMode
 
 
 @dataclass
@@ -65,31 +57,31 @@ CLIMATE_DESCRIPTIONS: dict[str, TuyaClimateEntityDescription] = {
     # https://developer.tuya.com/en/docs/iot/categorykt?id=Kaiuz0z71ov2n
     "kt": TuyaClimateEntityDescription(
         key="kt",
-        switch_only_hvac_mode=HVAC_MODE_COOL,
+        switch_only_hvac_mode=HVACMode.COOL,
     ),
     # Heater
     # https://developer.tuya.com/en/docs/iot/f?id=K9gf46epy4j82
     "qn": TuyaClimateEntityDescription(
         key="qn",
-        switch_only_hvac_mode=HVAC_MODE_HEAT,
+        switch_only_hvac_mode=HVACMode.HEAT,
     ),
     # Heater
     # https://developer.tuya.com/en/docs/iot/categoryrs?id=Kaiuz0nfferyx
     "rs": TuyaClimateEntityDescription(
         key="rs",
-        switch_only_hvac_mode=HVAC_MODE_HEAT,
+        switch_only_hvac_mode=HVACMode.HEAT,
     ),
     # Thermostat
     # https://developer.tuya.com/en/docs/iot/f?id=K9gf45ld5l0t9
     "wk": TuyaClimateEntityDescription(
         key="wk",
-        switch_only_hvac_mode=HVAC_MODE_HEAT_COOL,
+        switch_only_hvac_mode=HVACMode.HEAT_COOL,
     ),
     # Thermostatic Radiator Valve
     # Not documented
     "wkf": TuyaClimateEntityDescription(
         key="wkf",
-        switch_only_hvac_mode=HVAC_MODE_HEAT,
+        switch_only_hvac_mode=HVACMode.HEAT,
     ),
 }
 
@@ -212,14 +204,14 @@ class TuyaClimateEntity(TuyaEntity, ClimateEntity):
         if enum_type := self.find_dpcode(
             DPCode.MODE, dptype=DPType.ENUM, prefer_function=True
         ):
-            self._attr_hvac_modes = [HVAC_MODE_OFF]
+            self._attr_hvac_modes = [HVACMode.OFF]
             for tuya_mode, ha_mode in TUYA_HVAC_TO_HA.items():
                 if tuya_mode in enum_type.range:
                     self._hvac_to_tuya[ha_mode] = tuya_mode
                     self._attr_hvac_modes.append(ha_mode)
         elif self.find_dpcode(DPCode.SWITCH, prefer_function=True):
             self._attr_hvac_modes = [
-                HVAC_MODE_OFF,
+                HVACMode.OFF,
                 description.switch_only_hvac_mode,
             ]
 
@@ -283,9 +275,9 @@ class TuyaClimateEntity(TuyaEntity, ClimateEntity):
                         self.device.name,
                     )
 
-    def set_hvac_mode(self, hvac_mode: str) -> None:
+    def set_hvac_mode(self, hvac_mode: HVACMode) -> None:
         """Set new target hvac mode."""
-        commands = [{"code": DPCode.SWITCH, "value": hvac_mode != HVAC_MODE_OFF}]
+        commands = [{"code": DPCode.SWITCH, "value": hvac_mode != HVACMode.OFF}]
         if hvac_mode in self._hvac_to_tuya:
             commands.append(
                 {"code": DPCode.MODE, "value": self._hvac_to_tuya[hvac_mode]}
@@ -411,24 +403,24 @@ class TuyaClimateEntity(TuyaEntity, ClimateEntity):
         return round(self._set_humidity.scale_value(humidity))
 
     @property
-    def hvac_mode(self) -> str:
+    def hvac_mode(self) -> HVACMode:
         """Return hvac mode."""
         # If the switch off, hvac mode is off as well. Unless the switch
         # the switch is on or doesn't exists of course...
         if not self.device.status.get(DPCode.SWITCH, True):
-            return HVAC_MODE_OFF
+            return HVACMode.OFF
 
         if DPCode.MODE not in self.device.function:
             if self.device.status.get(DPCode.SWITCH, False):
                 return self.entity_description.switch_only_hvac_mode
-            return HVAC_MODE_OFF
+            return HVACMode.OFF
 
         if (
             mode := self.device.status.get(DPCode.MODE)
         ) is not None and mode in TUYA_HVAC_TO_HA:
             return TUYA_HVAC_TO_HA[mode]
 
-        return HVAC_MODE_OFF
+        return HVACMode.OFF
 
     @property
     def fan_mode(self) -> str | None:
@@ -461,7 +453,7 @@ class TuyaClimateEntity(TuyaEntity, ClimateEntity):
             return
 
         # Fake turn on
-        for mode in (HVAC_MODE_HEAT_COOL, HVAC_MODE_HEAT, HVAC_MODE_COOL):
+        for mode in (HVACMode.HEAT_COOL, HVACMode.HEAT, HVACMode.COOL):
             if mode not in self.hvac_modes:
                 continue
             self.set_hvac_mode(mode)
@@ -474,5 +466,5 @@ class TuyaClimateEntity(TuyaEntity, ClimateEntity):
             return
 
         # Fake turn off
-        if HVAC_MODE_OFF in self.hvac_modes:
-            self.set_hvac_mode(HVAC_MODE_OFF)
+        if HVACMode.OFF in self.hvac_modes:
+            self.set_hvac_mode(HVACMode.OFF)