python - 如何使用 ast（抽象语法树）模块获取节点的完整“路径”？

Question

我正在修改 Python 的ast（抽象语法树）模块。

使用ast.NodeVisitor，可以遍历源树并显示信息，例如每个节点的名称，如下所示：

class MyAST(ast.NodeVisitor):      

     def visit_ClassDef(self, node):
         print(node.col_offset * " " +node.name)         
         self.generic_visit(node)   

     def visit_FunctionDef(self, node):         
         print(node.col_offset * " " + node.name)     
         self.generic_visit(node)   

然后它可以用于任何这样的代码：

tree = ast.parse(open("/path/to/file.py").read())
MyAST().visit(tree)

这将输出类似的内容（例如，在requests包中的models.py上使用）：

RequestEncodingMixin                                                                                                                                                         
    path_url                                                                                                                                                                 
    _encode_params                                                                                                                                                           
    _encode_files                                                                                                                                                            
RequestHooksMixin                                                                                                                                                            
    register_hook                                                                                                                                                            
    deregister_hook                                                                                                                                                          
Request                                                                                                                                                                      
    __init__
    __repr__
    prepare
(...)

我想知道，对于给定的节点，它的完整“路径”。在上面的输出示例中，当我在FunctionDef节点中时deregister_hook，我想知道它的父节点是ClassDef RequestHooksMixin这样的，以便我可以输出类似RequestHooksMixin.deregister_hook.

标签： pythonabstract-syntax-tree