Как получить имена всех переменных, определенных в методах класса

Question

У меня есть следующий класс:

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

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


print Foo.setUp.__code__.co_varnames 
print Foo.bar.__code__.co_varnames 
print Foo.baz.__code__.co_varnames 
Output: 
('self',) 
('self', 'var3') 
('self', 'var', 'var5') 

Answer 2

Foo.__dict__ или vars(Foo) представляет собой словарь вида {ключ: атрибут, key2: atrribute2}

for k, v in Foo.__dict__.items(): 
    if hasattr(v, '__call__'): # type function check 
    items.append(k) # method name 
    items += v.__code__.co_names 
    for var in v.__code__.co_varnames: 
     if var != "self": 
      items.append(var) # var name 

print items 

выход

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

Answer 3

Не могу пока комментировать (не хватает репутации) , но это похоже на дубликат Iterate over object attributes in python.

Попробуйте это:

print([a for a in dir(obj) if not a.startswith('__') and not callable(getattr(obj,a))]) 

Answer 4

Вы можете использовать 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']