From 97446a5af3c7914d4ab40dddc3286f167f333c56 Mon Sep 17 00:00:00 2001
From: G Johansson <goran.johansson@shiftit.se>
Date: Sun, 4 Feb 2024 08:59:34 -0500
Subject: [PATCH] Add migrated climate feature flag to switchbee (#109458)

---
 homeassistant/components/switchbee/climate.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/homeassistant/components/switchbee/climate.py b/homeassistant/components/switchbee/climate.py
index 8dd740262f9..1fc5cfcba12 100644
--- a/homeassistant/components/switchbee/climate.py
+++ b/homeassistant/components/switchbee/climate.py
@@ -1,4 +1,5 @@
 """Support for SwitchBee climate."""
+
 from __future__ import annotations
 
 from typing import Any
@@ -87,11 +88,9 @@ async def async_setup_entry(
 class SwitchBeeClimateEntity(SwitchBeeDeviceEntity[SwitchBeeThermostat], ClimateEntity):
     """Representation of a SwitchBee climate."""
 
-    _attr_supported_features = (
-        ClimateEntityFeature.TARGET_TEMPERATURE | ClimateEntityFeature.FAN_MODE
-    )
     _attr_fan_modes = SUPPORTED_FAN_MODES
     _attr_target_temperature_step = 1
+    _enable_turn_on_off_backwards_compatibility = False
 
     def __init__(
         self,
@@ -106,6 +105,13 @@ class SwitchBeeClimateEntity(SwitchBeeDeviceEntity[SwitchBeeThermostat], Climate
         self._attr_temperature_unit = HVAC_UNIT_SB_TO_HASS[device.temperature_unit]
         self._attr_hvac_modes = [HVAC_MODE_SB_TO_HASS[mode] for mode in device.modes]
         self._attr_hvac_modes.append(HVACMode.OFF)
+        self._attr_supported_features = (
+            ClimateEntityFeature.TARGET_TEMPERATURE | ClimateEntityFeature.FAN_MODE
+        )
+        if len(self.hvac_modes) > 1:
+            self._attr_supported_features |= (
+                ClimateEntityFeature.TURN_OFF | ClimateEntityFeature.TURN_ON
+            )
         self._update_attrs_from_coordinator()
 
     @callback