在Python中,对ast.nodevisitor进行后序遍历

3
在Python中,通过操作ast.NodeVisitor.generic_visit(),可以对ast.NodeVisitor的实例进行后序遍历吗? 我尝试了以下代码:
class ExpParser(ast.NodeVisitor):

    def generic_visit(self, node):
        for x in ast.iter_child_nodes(node):
            ast.NodeVisitor.generic_visit(self, x)
        ast.NodeVisitor.generic_visit(self, node)

    def visit_BinOp(self, node):
        print type(node.op).__name__ 

    def visit_Name(self, node):
        print node.id

if __name__ == '__main__':
    node = ast.parse("T1+T2*T3")
    v = ExpParser()
    v.visit(node)

这让我得到了:
T1
T2
T3
Mult
Add

我希望它能为我提供以下内容:
T2
T3
Mult
T1
Add

我该如何做?我卡住了,请帮忙。

我注意到已弃用的compiler.visitor包有这个功能,但在ast包中似乎没有相应的功能。 - snim2
1个回答

3
尝试类似以下的内容:

尝试这样做

import ast

class ExpParser(ast.NodeVisitor):

    def generic_visit(self, node):
        for field, value in reversed(list(ast.iter_fields(node))):
            if isinstance(value, list):
                for item in value:
                    if isinstance(item, ast.AST):
                        self.visit(item)
            elif isinstance(value, ast.AST):
                self.visit(value)

    def visit_BinOp(self, node):
        self.generic_visit(node)
        print type(node.op).__name__ 

    def visit_Name(self, node):
        self.generic_visit(node)
        print node.id

if __name__ == '__main__':
    node = ast.parse("T1+T2*T3")
    v = ExpParser()
    v.visit(node)

不要忘记在每个访问方法中调用generic_visit函数,或者重新实现访问方法以自动处理它。

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接