python - 如何获取类方法中定义的所有变量的名称

Question

我有以下类(class):

class Foo(object):
    def setUp(self):
        self.var1 = "some value"
        self.var2 = something
    def bar(self):
        var3 = some value
    def baz(self, var):
        var4 = some value

我想打印方法中定义的所有变量的名称，例如:

setUp, bar, baz, var1, var2, var3, var4

我试过使用 locals()、vars()、globals()，但我只得到方法的名称，而不是变量名称。

我也尝试过使用 ast 模块，但没有成功。

Answer 1

您可以使用 ast.parse生成AST节点。那么ast.walk可用于递归迭代节点及其后代。对于每个节点，您可以检查类型，然后提取正确的属性。

下面是一个使用您的代码的示例，但不要指望它能像这样处理更复杂的文件:

source = '''
class Foo(object):
    def setUp(self):
        self.var1 = "some value"
        self.var2 = 1
    def bar(self):
        var3 = 2
    def baz(self, var):
        var4 = var
'''

import ast

def hack(source):
    root = ast.parse(source)

    for node in ast.walk(root):
        if isinstance(node, ast.Name) and isinstance(node.ctx, ast.Store):
            yield node.id
        elif isinstance(node, ast.Attribute):
            yield node.attr
        elif isinstance(node, ast.FunctionDef):
            yield node.name

print(list(hack(source)))

输出:

['setUp', 'bar', 'baz', 'var1', 'var2', 'var3', 'var4']