Question

I m tinkering with Zhu ast(Abstractyntax大树s)模块。

利用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)

这将产生一些成果(例如:models.py,从requests包装:

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

我愿就某一节点知道对它的全部“pa”。在上述产出样本中,如果在<条码>功能设计/代码>上出现I m,则我知道其母是<条码>。职等DefRequestHooksMixin,以便我能产生RequestHooksMixin.deregister_hook<>。

Answer 1


ast.NodeVisitor 设计时,没有考虑到加固等级,因此,它是产生基于等级的缩减的产出的最佳工具。

相反,你可以使用<条码>ast.iter_child_nodes,以 rec击每一节日子的直接儿童节点,并在课堂或职能时,产生节点和增加退学水平。

import ast

def build_tree(node):
    def _visit(node, level=0):
        if isinstance(node, (ast.ClassDef, ast.FunctionDef)):
            yield node.name, level
            level += 1
        for child in ast.iter_child_nodes(node):
            yield from _build_tree(child, level)
    return  
 .join(f {"    " * level}{name}  for name, level in _visit(node))

so that:

source =    
class Foo:
    def method1(self):
        pass
    def method2(self):
        pass

class Bar:
    def method1(self):
        pass
    def method2(self):
        pass
   
print(build_tree(ast.parse(source)))

outputs:

Foo
    method1
    method2
Bar
    method1
    method2

Demo:

友情链接