Question

How to list all methods/classes that have been imported into a class. Take an example

from module1 import method1
from module2 import method2

class foo(object):
   def say_foo(self):
      method1()
   def talk(self):
      method2()

is it possible to list all methods used in the foo class? i know inspect.getmembers(foo) will list talk and say_foo, how do i list method1 and method2?

Answer 1

use ast module to analyze you code:

code = """
from module1 import method1
from module2 import method2

class foo(object):
   def say_foo(self):
      method1()
   def talk(self):
      method2()
"""

import ast, _ast

t = ast.parse(code)
for node in ast.walk(t):
    if isinstance(node, ast.ClassDef) and node.name == "foo":
        klass = node
        break

for node in ast.walk(klass):
    if isinstance(node, _ast.Call):
        print node.func.id

the output is :

method1
method2