unroller.py 文件源码

def sub_subscript(root, subs):
    root = copy.deepcopy(root)

    class Transformer(ast.NodeTransformer):
        def visit_FunctionDef(self, node):
            return node

        def visit_Subscript(self, node):
            self.generic_visit(node)

            try:
                node_tup = subscript_to_tuple(node)
                if node_tup in subs:
                    return subs[node_tup]
                else:
                    return node

            except ValueError:
                return node

    return Transformer().visit(root)