zewt · April 3, 2018 04:02 · Apr 3, 2018
diff --git a/gistfile1.txt b/gistfile1.txt
@@ -0,0 +1,32 @@
+import sys
+import pymel.core as pm
+
+def pick_walk(direction, add=False):
+    nodes = pm.ls(sl=True)
+    new_nodes = []
+    non_dag_nodes = []
+    for node in nodes:
+        # Make a list of non-DAG nodes.
+        if not isinstance(node, pm.nodetypes.DagNode):
+            non_dag_nodes.append(node)
+            continue
+
+        pm.select(node, ne=True)
+        pm.pickWalk(direction=direction)
+        new_nodes.extend(pm.ls(sl=True))
+
+    pm.select(d=True)
+
+    # If any non-DAG objects were selected, like components, use the regular pick walk command to handle them.
+    if non_dag_nodes:
+        pm.select(non_dag_nodes)
+        pm.pickWalk(direction=direction)
+
+    # Add our individual pick walking results.
+    pm.select(new_nodes) #, ne=True, add=True)
+
+    # If add is true, reselect the original selection.
+    if add:
+        pm.select(nodes, ne=True, add=True)
+
+pick_walk('down', False)