Python AST处理

Question

我有一个由ast.parse()返回的Python AST。

我知道这是一个类方法的AST。

如何找到对同一个类的其他方法的所有调用？

基本上，我想收集一些东西，比如：

['foo', 'bar']

对于像这样的代码片段：

def baz(self): # this is a class method
    '''baz docstring'''
    self.foo() + self.bar()

我需要一个函数，它将接受AST并将其他方法的方法名列表作为类的字符串返回，这些字符串在同一个类的方法中被调用。

Answer 1

一般的方法是继承ast.NodeVisitor的子类

>>> class VisitCalls(ast.NodeVisitor):
...   def visit_Call(self, what):
...     if what.func.value.id == 'self':
...       print what.func.attr
... 
>>> f='''def x(self):
...   return self.bar() + self.baz()
... '''
>>> xx = ast.parse(f)
>>> VisitCalls().visit(xx)
bar
baz

但是，这只会捕获对self.something的“立即”调用。在一般情况下，你可以使用somelist.append(self.blah)，然后很久以后在代码somelist[i + j]()中:确定后者是对self.blah的调用还是对与当前实例的方法无关的其他可调用的调用的问题是图灵完成(CS术语表示“在一般情况下完全不可解”，就像数学家可能会说的"NP-hard";-)。

但是，如果您所需要的只是解决简单的“立即呼叫”情况，那么您就可以继续了；-)。