Aborted Maya pick walking script

gistfile1.txt

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)