Add tuya switch_mode for rotary knobs

Author: elupus

zha/application/platforms/select.py

@@ -14,6 +14,7 @@
     TUYA_PLUG_MANUFACTURER,
     TUYA_PLUG_ONOFF,
 )
+from zhaquirks import tuya
 from zhaquirks.xiaomi.aqara.magnet_ac01 import OppleCluster as MagnetAC01OppleCluster
 from zhaquirks.xiaomi.aqara.switch_acn047 import OppleCluster as T2RelayOppleCluster
 from zigpy import types
@@ -317,6 +318,18 @@ class TuyaBacklightModeSelectEntity(ZCLEnumSelectEntity):
     _attr_translation_key: str = "backlight_mode"
 
 
+@CONFIG_DIAGNOSTIC_MATCH(
+    cluster_handler_names=CLUSTER_HANDLER_ON_OFF, exposed_features=TUYA_PLUG_ONOFF
+)
+class TuyaSwitchModeSelectEntity(ZCLEnumSelectEntity):
+    """Representation of a ZHA backlight mode select entity."""
+
+    _unique_id_suffix = "switch_mode"
+    _attribute_name = "switch_mode"
+    _enum = tuya.SwitchMode
+    _attr_translation_key: str = "switch_mode"
+
+
 class MoesBacklightMode(types.enum8):
     """MOES switch backlight mode enum."""
 

zha/application/platforms/switch.py

@@ -111,6 +111,7 @@ class Switch(PlatformEntity, BaseSwitch):
 
     _attr_translation_key = "switch"
     _attr_primary_weight = 10
+    _attribute_name = OnOff.AttributeDefs.on_off.name
 
     def __init__(
         self,
@@ -135,12 +136,29 @@ def on_add(self) -> None:
             )
         )
 
+    def _is_supported(self) -> bool:
+        if (
+            self._attribute_name
+            in self._on_off_cluster_handler.cluster.unsupported_attributes
+            or self._attribute_name
+            not in self._on_off_cluster_handler.cluster.attributes_by_name
+            or self._on_off_cluster_handler.cluster.get(self._attribute_name) is None
+        ):
+            _LOGGER.debug(
+                "%s is not supported - skipping %s entity creation",
+                self._attribute_name,
+                self.__class__.__name__,
+            )
+            return False
+
+        return super()._is_supported()
+
     def handle_cluster_handler_attribute_updated(
         self,
         event: ClusterAttributeUpdatedEvent,  # pylint: disable=unused-argument
     ) -> None:
         """Handle state update from cluster handler."""
-        if event.attribute_name == OnOff.AttributeDefs.on_off.name:
+        if event.attribute_name == self._attribute_name:
             self.maybe_emit_state_changed_event()
 
 

zha/zigbee/cluster_handlers/general.py

@@ -550,6 +550,7 @@ def __init__(self, cluster: zigpy.zcl.Cluster, endpoint: Endpoint) -> None:
             self.ZCL_INIT_ATTRS["backlight_mode"] = True
             self.ZCL_INIT_ATTRS["power_on_state"] = True
             self.ZCL_INIT_ATTRS["child_lock"] = True
+            self.ZCL_INIT_ATTRS["switch_mode"] = True
 
     @classmethod
     def matches(cls, cluster: zigpy.zcl.Cluster, endpoint: Endpoint) -> bool: