loop_parallel_ast.py 文件源码

def get_index(node):
    """
    get the index followed after the given namenode if possible
    """

    parent = node.parent
    if isinstance(parent, ast.Subscript):
        try:
            ind_node = parent.slice.value
            #if there's multiple dimensions in index
            if isinstance(ind_node, ast.Tuple):
                value = []
                for i in range(len(ind_node.elts)):
                    ind_value = py_ast.dump_ast(ind_node.elts[i])
                    value.append(ind_value)
                return value
            else:
                value = py_ast.dump_ast(ind_node)
                return value
        except:
            try:
                value = []
                ind_node = parent.slice.dims
                for i in range(len(ind_node)):
                    ind_value = py_ast.dump_ast(ind_node[i])
                    value.append(ind_value)
                return value
            except:
                return []
    else:
        return []