Add support for additional Legrand wireless switches (#4114)

Author: nzapponi

zhaquirks/legrand/wirelessswitch.py

@@ -1,15 +1,21 @@
-"""Module for Legrand wireless radiant switch."""
+"""Module for Legrand wireless switches (Radiant, NLD, NLT, NLW)."""
 
 from zigpy.quirks.v2 import QuirkBuilder
 from zigpy.zcl.clusters.general import BinaryInput
 
 from zhaquirks.const import (
     BUTTON,
+    BUTTON_1,
+    BUTTON_2,
+    BUTTON_3,
     COMMAND,
     COMMAND_MOVE,
     COMMAND_OFF,
     COMMAND_ON,
     COMMAND_STOP,
+    DIM_DOWN,
+    DIM_UP,
+    ENDPOINT_ID,
     LONG_PRESS,
     LONG_RELEASE,
     PARAMS,
@@ -40,3 +46,90 @@
     )
     .add_to_registry()
 )
+
+(
+    QuirkBuilder(f" {LEGRAND}", " NLWO - Triple remote switch")
+    .replaces(LegrandPowerConfigurationCluster)
+    .prevent_default_entity_creation(endpoint_id=1, cluster_id=BinaryInput.cluster_id)
+    .device_automation_triggers(
+        {
+            (TURN_ON, BUTTON_1): {COMMAND: COMMAND_ON, ENDPOINT_ID: 1},
+            (DIM_UP, BUTTON_1): {
+                COMMAND: COMMAND_MOVE,
+                ENDPOINT_ID: 1,
+                PARAMS: {"move_mode": 0, "rate": 255},
+            },
+            (TURN_OFF, BUTTON_1): {COMMAND: COMMAND_OFF},
+            (DIM_DOWN, BUTTON_1): {
+                COMMAND: COMMAND_MOVE,
+                ENDPOINT_ID: 1,
+                PARAMS: {"move_mode": 1, "rate": 255},
+            },
+            (LONG_RELEASE, BUTTON_1): {COMMAND: COMMAND_STOP, ENDPOINT_ID: 1},
+            (TURN_ON, BUTTON_2): {COMMAND: COMMAND_ON, ENDPOINT_ID: 2},
+            (DIM_UP, BUTTON_2): {
+                COMMAND: COMMAND_MOVE,
+                ENDPOINT_ID: 2,
+                PARAMS: {"move_mode": 0, "rate": 255},
+            },
+            (TURN_OFF, BUTTON_2): {COMMAND: COMMAND_OFF},
+            (DIM_DOWN, BUTTON_2): {
+                COMMAND: COMMAND_MOVE,
+                ENDPOINT_ID: 2,
+                PARAMS: {"move_mode": 1, "rate": 255},
+            },
+            (LONG_RELEASE, BUTTON_2): {COMMAND: COMMAND_STOP, ENDPOINT_ID: 2},
+            (TURN_ON, BUTTON_3): {COMMAND: COMMAND_ON, ENDPOINT_ID: 3},
+            (DIM_UP, BUTTON_3): {
+                COMMAND: COMMAND_MOVE,
+                ENDPOINT_ID: 3,
+                PARAMS: {"move_mode": 0, "rate": 255},
+            },
+            (TURN_OFF, BUTTON_3): {COMMAND: COMMAND_OFF},
+            (DIM_DOWN, BUTTON_3): {
+                COMMAND: COMMAND_MOVE,
+                ENDPOINT_ID: 3,
+                PARAMS: {"move_mode": 1, "rate": 255},
+            },
+            (LONG_RELEASE, BUTTON_3): {COMMAND: COMMAND_STOP, ENDPOINT_ID: 3},
+        }
+    )
+    .add_to_registry()
+)
+
+(
+    QuirkBuilder(f" {LEGRAND}", " Double gangs remote switch")
+    .replaces(LegrandPowerConfigurationCluster)
+    .prevent_default_entity_creation(endpoint_id=1, cluster_id=BinaryInput.cluster_id)
+    .device_automation_triggers(
+        {
+            (TURN_ON, BUTTON_1): {COMMAND: COMMAND_ON, ENDPOINT_ID: 1},
+            (DIM_UP, BUTTON_1): {
+                COMMAND: COMMAND_MOVE,
+                ENDPOINT_ID: 1,
+                PARAMS: {"move_mode": 0, "rate": 255},
+            },
+            (TURN_OFF, BUTTON_1): {COMMAND: COMMAND_OFF},
+            (DIM_DOWN, BUTTON_1): {
+                COMMAND: COMMAND_MOVE,
+                ENDPOINT_ID: 1,
+                PARAMS: {"move_mode": 1, "rate": 255},
+            },
+            (LONG_RELEASE, BUTTON_1): {COMMAND: COMMAND_STOP, ENDPOINT_ID: 1},
+            (TURN_ON, BUTTON_2): {COMMAND: COMMAND_ON, ENDPOINT_ID: 2},
+            (DIM_UP, BUTTON_2): {
+                COMMAND: COMMAND_MOVE,
+                ENDPOINT_ID: 2,
+                PARAMS: {"move_mode": 0, "rate": 255},
+            },
+            (TURN_OFF, BUTTON_2): {COMMAND: COMMAND_OFF},
+            (DIM_DOWN, BUTTON_2): {
+                COMMAND: COMMAND_MOVE,
+                ENDPOINT_ID: 2,
+                PARAMS: {"move_mode": 1, "rate": 255},
+            },
+            (LONG_RELEASE, BUTTON_2): {COMMAND: COMMAND_STOP, ENDPOINT_ID: 2},
+        }
+    )
+    .add_to_registry()
+)