Updated

Scripts/Animation/epic_pose_wrangler/v2/model/actions/selection.py

@@ -0,0 +1,52 @@
+#  Copyright Epic Games, Inc. All Rights Reserved.
+from maya import cmds
+
+from epic_pose_wrangler.v2.model import base_action
+
+class SelectSolverAction(base_action.BaseAction):
+    __display_name__ = "Select Solver Node(s)"
+    __tooltip__ = "Selects the currently selected solver nodes in the scene"
+    __category__ = "Select"
+
+    @classmethod
+    def validate(cls, ui_context):
+        return bool(ui_context.current_solvers)
+
+    def execute(self, ui_context=None, **kwargs):
+        if not ui_context:
+            ui_context = self.api.get_ui_context()
+        if not ui_context:
+            return
+        cmds.select(ui_context.current_solvers, replace=True)
+
+class SelectDriverAction(base_action.BaseAction):
+    __display_name__ = "Select Driver Node(s)"
+    __tooltip__ = "Selects the driver nodes in the scene"
+    __category__ = "Select"
+
+    @classmethod
+    def validate(cls, ui_context):
+        return bool(ui_context.drivers)
+
+    def execute(self, ui_context=None, **kwargs):
+        if not ui_context:
+            ui_context = self.api.get_ui_context()
+        if not ui_context:
+            return
+        cmds.select(ui_context.drivers, replace=True)
+
+class SelectDrivenAction(base_action.BaseAction):
+    __display_name__ = "Select Driven Node(s)"
+    __tooltip__ = "Selects the driven nodes in the scene"
+    __category__ = "Select"
+
+    @classmethod
+    def validate(cls, ui_context):
+        return bool(ui_context.driven)
+
+    def execute(self, ui_context=None, **kwargs):
+        if not ui_context:
+            ui_context = self.api.get_ui_context()
+        if not ui_context:
+            return
+        cmds.select(ui_context.driven, replace=True)