如何使用ast.NodeVisitor的简单示例？

Question

有谁有一个使用 ast.NodeVisitor 来遍历 Python 2.6 中的抽象语法树的简单示例吗？我不清楚访问和 generic_visit 之间的区别，而且我找不到任何使用谷歌代码搜索或普通谷歌的示例。

Answer 1

ast.visit ——当然，除非你在子类中重写它 ——当调用访问类 foo 的 ast.Node 时，如果该方法存在，则调用 self.visit_foo ，否则调用 self.generic_visit 。后者在类 ast 本身的实现中，仅在每个子节点上调用 self.visit （并且不执行其他操作）。

因此，请考虑，例如：

>>> class v(ast.NodeVisitor):
...   def generic_visit(self, node):
...     print type(node).__name__
...     ast.NodeVisitor.generic_visit(self, node)
... 

在这里，我们重写 generic_visit 来打印类名，但也调用基类（以便所有子类也将被调用）访问过）。例如...：

>>> x = v()
>>> t = ast.parse('d[x] += v[y, x]')
>>> x.visit(t)

发出：

Module
AugAssign
Subscript
Name
Load
Index
Name
Load
Store
Add
Subscript
Name
Load
Index
Tuple
Name
Load
Name
Load
Load
Load

但是假设我们不关心负载节点（及其子节点——如果它们有的话；-）。那么处理这个问题的简单方法可能是，例如：

>>> class w(v):
...   def visit_Load(self, node): pass
... 

现在，当我们访问 Load 节点时，visit 分派，不再是 generic_visit ，而是分派到我们的新节点visit_Load...它根本不执行任何操作。所以：

>>> y = w()
>>> y.visit(t)
Module
AugAssign
Subscript
Name
Index
Name
Store
Add
Subscript
Name
Index
Tuple
Name
Name

或者，假设我们还想查看名称节点的实际名称； then...：

>>> class z(v):
...   def visit_Name(self, node): print 'Name:', node.id
... 
>>> z().visit(t)
Module
AugAssign
Subscript
Name: d
Index
Name: x
Store
Add
Subscript
Name: v
Index
Tuple
Name: y
Name: x
Load
Load

但是，NodeVisitor 是一个类，因为这可以让它在访问期间存储信息。假设我们想要的只是“模块”中的一组名称。那么我们就不再需要重写 generic_visit 了，而是......：

>>> class allnames(ast.NodeVisitor):
...   def visit_Module(self, node):
...     self.names = set()
...     self.generic_visit(node)
...     print sorted(self.names)
...   def visit_Name(self, node):
...     self.names.add(node.id)
... 
>>> allnames().visit(t)
['d', 'v', 'x', 'y']

与需要重写 generic_visit 的用例相比，这种事情是更典型的用例 --通常，您只对几种节点感兴趣，就像我们在 Module 和 Name 中一样，因此我们可以覆盖 visit_Module 和 visit_Name 并让 ast 的 visit 代表我们进行调度。

Answer 2

查看 ast.py 中的代码并不难复制粘贴并滚动你自己的助行器。例如

import ast
def str_node(node):
    if isinstance(node, ast.AST):
        fields = [(name, str_node(val)) for name, val in ast.iter_fields(node) if name not in ('left', 'right')]
        rv = '%s(%s' % (node.__class__.__name__, ', '.join('%s=%s' % field for field in fields))
        return rv + ')'
    else:
        return repr(node)
def ast_visit(node, level=0):
    print('  ' * level + str_node(node))
    for field, value in ast.iter_fields(node):
        if isinstance(value, list):
            for item in value:
                if isinstance(item, ast.AST):
                    ast_visit(item, level=level+1)
        elif isinstance(value, ast.AST):
            ast_visit(value, level=level+1)


ast_visit(ast.parse('a + b'))

打印出来

Module(body=[<_ast.Expr object at 0x02808510>])
  Expr(value=BinOp(op=Add()))
    BinOp(op=Add())
      Name(id='a', ctx=Load())
        Load()
      Add()
      Name(id='b', ctx=Load())
        Load()

Answer 3

当找不到自定义访问者（即visit_Name）时，将调用generic_visit。这是我最近用 ast.NodeVisitor 编写的一段代码： https ://foss.heptapod.net/pypy/pypy/-/blob/80ead76ab428100ffeb01109c7fc0d94f1048af2/py/_code/_assertionnew.py 它解释 AST 节点以获取有关其中一些节点的调试信息，并使用 generic_visit 当未提供特殊实现时。