Convert Sonoff buttons to v2 quirk (#3050)

* Add Sonoff button `SNZB-01P` variant

* Inherit `SonoffSmartButtonSNZB01P` to use existing device triggers

* Initial v2 quirk conversion for Sonoff button

* Remove signature test for Sonoff button

* Also convert other Sonoff button to v2

* Merge Sonoff button quirks into one file

* Combine both v2 quirks into one, use less strict device triggers

Author: TheJulianJES

tests/test_sonoff.py

@@ -1,70 +1,25 @@
-"""Device handler for eWeLink WB01."""
-from zigpy.profiles import zha
-from zigpy.quirks import CustomDevice
-from zigpy.zcl.clusters.general import Basic, Identify, OnOff, PowerConfiguration
+"""Device handler for Sonoff buttons."""
+from zigpy.quirks.v2 import add_to_registry_v2
 
 from zhaquirks.const import (
     BUTTON,
     COMMAND,
     COMMAND_OFF,
     COMMAND_ON,
     COMMAND_TOGGLE,
-    DEVICE_TYPE,
     DOUBLE_PRESS,
-    ENDPOINTS,
-    INPUT_CLUSTERS,
     LONG_PRESS,
-    MODELS_INFO,
-    OUTPUT_CLUSTERS,
-    PROFILE_ID,
     SHORT_PRESS,
 )
 
-
-class SonoffButton(CustomDevice):
-    """Custom device representing sonoff devices."""
-
-    signature = {
-        # <SimpleDescriptor endpoint=1 profile=260 device_type=0 device_version=0
-        # input_clusters=[0, 3, 1] output_clusters=[6, 3]>
-        MODELS_INFO: [("eWeLink", "WB01")],
-        ENDPOINTS: {
-            1: {
-                PROFILE_ID: zha.PROFILE_ID,
-                DEVICE_TYPE: zha.DeviceType.ON_OFF_SWITCH,
-                INPUT_CLUSTERS: [
-                    Basic.cluster_id,
-                    PowerConfiguration.cluster_id,
-                    Identify.cluster_id,
-                ],
-                OUTPUT_CLUSTERS: [
-                    Identify.cluster_id,
-                    OnOff.cluster_id,
-                ],
-            },
-        },
-    }
-
-    replacement = {
-        ENDPOINTS: {
-            1: {
-                PROFILE_ID: zha.PROFILE_ID,
-                DEVICE_TYPE: zha.DeviceType.ON_OFF_SWITCH,
-                INPUT_CLUSTERS: [
-                    Basic.cluster_id,
-                    PowerConfiguration.cluster_id,
-                    Identify.cluster_id,
-                ],
-                OUTPUT_CLUSTERS: [
-                    Identify.cluster_id,
-                    OnOff.cluster_id,
-                ],
-            },
+(
+    add_to_registry_v2("eWeLink", "WB01")
+    .also_applies_to("eWeLink", "SNZB-01P")
+    .device_automation_triggers(
+        {
+            (SHORT_PRESS, BUTTON): {COMMAND: COMMAND_TOGGLE},
+            (DOUBLE_PRESS, BUTTON): {COMMAND: COMMAND_ON},
+            (LONG_PRESS, BUTTON): {COMMAND: COMMAND_OFF},
         }
-    }
-
-    device_automation_triggers = {
-        (SHORT_PRESS, BUTTON): {COMMAND: COMMAND_TOGGLE},
-        (DOUBLE_PRESS, BUTTON): {COMMAND: COMMAND_ON},
-        (LONG_PRESS, BUTTON): {COMMAND: COMMAND_OFF},
-    }
+    )
+)