Signature code clean up for TS0003 Tuya quirk (#2591)

zigpy · Sep 22, 2023 · cfca1b8 · cfca1b8
1 parent 7745c83
commit cfca1b8
Showing 1 changed file with 14 additions and 14 deletions.
diff --git a/zhaquirks/tuya/ts000x.py b/zhaquirks/tuya/ts000x.py
@@ -653,8 +653,8 @@ class Switch_3G_GPP_Var2(EnchantedDevice):
                     Groups.cluster_id,
                     Scenes.cluster_id,
                     OnOff.cluster_id,
-                    0xE000,
-                    0xE001,
+                    TuyaZBE000Cluster.cluster_id,
+                    TuyaZBExternalSwitchTypeCluster.cluster_id,
                 ],
                 OUTPUT_CLUSTERS: [
                     Time.cluster_id,
@@ -669,8 +669,8 @@ class Switch_3G_GPP_Var2(EnchantedDevice):
                     Groups.cluster_id,
                     Scenes.cluster_id,
                     OnOff.cluster_id,
-                    0xE000,
-                    0xE001,
+                    TuyaZBE000Cluster.cluster_id,
+                    TuyaZBExternalSwitchTypeCluster.cluster_id,
                 ],
                 OUTPUT_CLUSTERS: [],
             },
@@ -682,8 +682,8 @@ class Switch_3G_GPP_Var2(EnchantedDevice):
                     Groups.cluster_id,
                     Scenes.cluster_id,
                     OnOff.cluster_id,
-                    0xE000,
-                    0xE001,
+                    TuyaZBE000Cluster.cluster_id,
+                    TuyaZBExternalSwitchTypeCluster.cluster_id,
                 ],
                 OUTPUT_CLUSTERS: [],
             },
@@ -1040,8 +1040,8 @@ class Switch_4G_GPP_Var2(EnchantedDevice):
                     Groups.cluster_id,
                     Scenes.cluster_id,
                     OnOff.cluster_id,
-                    0xE000,
-                    0xE001,
+                    TuyaZBE000Cluster.cluster_id,
+                    TuyaZBExternalSwitchTypeCluster.cluster_id,
                 ],
                 OUTPUT_CLUSTERS: [
                     Time.cluster_id,
@@ -1056,8 +1056,8 @@ class Switch_4G_GPP_Var2(EnchantedDevice):
                     Groups.cluster_id,
                     Scenes.cluster_id,
                     OnOff.cluster_id,
-                    0xE000,
-                    0xE001,
+                    TuyaZBE000Cluster.cluster_id,
+                    TuyaZBExternalSwitchTypeCluster.cluster_id,
                 ],
                 OUTPUT_CLUSTERS: [],
             },
@@ -1069,8 +1069,8 @@ class Switch_4G_GPP_Var2(EnchantedDevice):
                     Groups.cluster_id,
                     Scenes.cluster_id,
                     OnOff.cluster_id,
-                    0xE000,
-                    0xE001,
+                    TuyaZBE000Cluster.cluster_id,
+                    TuyaZBExternalSwitchTypeCluster.cluster_id,
                 ],
                 OUTPUT_CLUSTERS: [],
             },
@@ -1082,8 +1082,8 @@ class Switch_4G_GPP_Var2(EnchantedDevice):
                     Groups.cluster_id,
                     Scenes.cluster_id,
                     OnOff.cluster_id,
-                    0xE000,
-                    0xE001,
+                    TuyaZBE000Cluster.cluster_id,
+                    TuyaZBExternalSwitchTypeCluster.cluster_id,
                 ],
                 OUTPUT_CLUSTERS: [],
             },