chore: Remove use of decorator and replace with class attribute

Author: allenporter

roborock/devices/traits/v1/child_lock.py

@@ -5,11 +5,11 @@
 _STATUS_PARAM = "lock_status"
 
 
-@common.requires_feature("is_set_child_supported")
 class ChildLockTrait(ChildLockStatus, common.V1TraitMixin):
     """Trait for controlling the child lock of a Roborock device."""
 
     command = RoborockCommand.GET_CHILD_LOCK_STATUS
+    requires_feature = "is_set_child_supported"    
 
     async def enable(self) -> None:
         """Enable the child lock."""

roborock/devices/traits/v1/common.py

@@ -147,29 +147,3 @@ def wrapper(*args, **kwargs):
 
     cls.map_rpc_channel = True  # type: ignore[attr-defined]
     return wrapper
-
-
-def requires_feature(feature_attribute_name: str):
-    """Decorator to mark a trait as requiring a specific device feature.
-
-    This decorator marks a trait class as requiring a specific device feature
-    to be supported. The feature check should be a direct reference to a
-    DeviceFeatures attribute.
-
-    Args:
-        feature_attribute: Direct reference to the DeviceFeatures attribute
-                          (e.g., DeviceFeatures.is_set_child_supported)
-
-    Example:
-        @requires_feature("is_flow_led_setting_supported")
-        class FlowLedStatusTrait(FlowLedStatus, V1TraitMixin):
-            ...
-    """
-
-    def decorator(cls):
-        class Wrapper(cls):  # type: ignore[valid-type, misc]
-            requires_feature = feature_attribute_name  # type: ignore[attr-defined]
-
-        return Wrapper
-
-    return decorator

roborock/devices/traits/v1/flow_led_status.py

@@ -5,11 +5,11 @@
 _STATUS_PARAM = "status"
 
 
-@common.requires_feature("is_flow_led_setting_supported")
 class FlowLedStatusTrait(FlowLedStatus, common.V1TraitMixin):
     """Trait for controlling the Flow LED status of a Roborock device."""
 
     command = RoborockCommand.GET_FLOW_LED_STATUS
+    requires_feature = "is_flow_led_setting_supported"
 
     async def enable(self) -> None:
         """Enable the Flow LED status."""

roborock/devices/traits/v1/led_status.py

@@ -5,11 +5,11 @@
 from .common import V1ResponseData
 
 
-@common.requires_feature("is_led_status_switch_supported")
 class LedStatusTrait(LedStatus, common.V1TraitMixin):
     """Trait for controlling the LED status of a Roborock device."""
 
     command = RoborockCommand.GET_LED_STATUS
+    requires_feature = "is_led_status_switch_supported"
 
     async def enable(self) -> None:
         """Enable the LED status."""

roborock/devices/traits/v1/valley_electricity_timer.py

@@ -5,11 +5,11 @@
 _ENABLED_PARAM = "enabled"
 
 
-@common.requires_feature("is_supported_valley_electricity")
 class ValleyElectricityTimerTrait(ValleyElectricityTimer, common.V1TraitMixin):
     """Trait for managing Valley Electricity Timer settings on Roborock devices."""
 
     command = RoborockCommand.GET_VALLEY_ELECTRICITY_TIMER
+    requires_feature = "is_supported_valley_electricity"
 
     async def set_timer(self, timer: ValleyElectricityTimer) -> None:
         """Set the Valley Electricity Timer settings of the device."""