Fix Hass 2024.2 climate entity deprecation #1341

AlexxIT · Feb 12, 2024 · aa88006 · aa88006
1 parent 21bd38a
commit aa88006
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/custom_components/sonoff/climate.py b/custom_components/sonoff/climate.py
@@ -3,7 +3,7 @@
     ClimateEntityFeature,
     HVACMode,
 )
-from homeassistant.const import UnitOfTemperature
+from homeassistant.const import UnitOfTemperature, MAJOR_VERSION, MINOR_VERSION
 
 from .core.const import DOMAIN
 from .core.entity import XEntity
@@ -29,12 +29,22 @@ class XClimateTH(XEntity, ClimateEntity):
     _attr_hvac_modes = [HVACMode.OFF, HVACMode.HEAT, HVACMode.COOL, HVACMode.DRY]
     _attr_max_temp = 99
     _attr_min_temp = 1
-    _attr_supported_features = ClimateEntityFeature.TARGET_TEMPERATURE_RANGE
     _attr_target_temperature_high = None
     _attr_target_temperature_low = None
     _attr_temperature_unit = UnitOfTemperature.CELSIUS
     _attr_target_temperature_step = 1
 
+    # https://developers.home-assistant.io/blog/2024/01/24/climate-climateentityfeatures-expanded
+    if (MAJOR_VERSION, MINOR_VERSION) >= (2024, 2):
+        _attr_supported_features = (
+            ClimateEntityFeature.TARGET_TEMPERATURE_RANGE
+            | ClimateEntityFeature.TURN_ON
+            | ClimateEntityFeature.TURN_OFF
+        )
+        _enable_turn_on_off_backwards_compatibility = False
+    else:
+        _attr_supported_features = ClimateEntityFeature.TARGET_TEMPERATURE_RANGE
+
     heat: bool = None
 
     def set_state(self, params: dict):