Python AST 处理

发布于 2024-08-23 14:44:58 字数 363 浏览 5 评论 0原文

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

我知道这是类方法的 AST。

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

基本上，我想收集类似的内容：

['foo', 'bar']

对于这样的代码片段：

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

我需要一个函数，该函数将接受 AST 并返回在该方法的方法中调用的类的其他方法[字符串形式的方法名称]的列表同一个班级。

原文

I have a Python AST [as returned by ast.parse()].

I know this is an AST of a class method.

How do I find all calls to other methods of the same class?

Basically, I want to collect something like:

['foo', 'bar']

for a code snippet like:

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

I need a function that will accept an AST and will return the list of other methods [method names as strings] of the class that are being invoked inside a method of the same class.

分享到QQ

分享到微博

如果你对这篇内容有疑问，欢迎到本站社区发帖提问参与讨论，获取更多帮助，或者扫码二维码加入 Web 技术交流群。

发布评论

需要登录才能够评论，你可以免费注册一个本站的账号。

情愿 2024-08-30 14:44:58

一般方法是子类 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”一样；-)。

但如果您需要的只是解决简单的“立即呼叫”案例，那么您就可以开始了；-)。

The general approach is to subclass 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

However, this will only catch "immediate" calls to self.something. In the general case you could have e.g. somelist.append(self.blah) and then much later in the code somelist[i + j](): the problem of determining whether the latter is a call to self.blah or to some other callable that has nothing to do with methods of the current instance is Turing-complete (CS jargon for "completely insoluble in the general case", much like a mathematician might say "NP-hard";-).

But if all you need is to solve the simple "immediate call" case, you're good to go;-).

回复收藏 0 原文

~没有更多了~