從方法對象獲取方法調用及其參數

Question

使用pythons inspect模塊我已經隔離了一個方法對象，現在我需要遍歷方法中的源代碼以查找對某些其他方法的調用並獲取它們的參數。

例如，假設下面的類：

def my_method(): 
    print('hello') 
    foobar('apple', 'pear', 6) 
    print('world') 
    foobar(1, 2, 3) 
    return foobar('a', 'b') 

我需要提取傳遞給foobar()的參數列表：

[('apple', 'pear', 6), (1, 2, 3), ('a', 'b', None)] 

可以假定所有的參數都是硬編碼，而不是動態。

給定inspect包中的方法對象，如何檢查所述方法中的方法調用？

嘗試

• 我一直在使用正則表達式試圖用inspect.getsourcelines(method)但是這打破如果參數語法的變化。
• 我已經用pythons ast模塊研究了抽象語法樹，但沒有找到任何解決方案。
• 必須有一種方法來完成這個使用inspect但我再也沒有找到任何解決方案。

Answer 1

這是不完美的，但應該是一個開始，我會在個位相加更好的實施方案：

from ast import parse, Call, walk 
import importlib 
import inspect 

mod = "test" 
mod = importlib.import_module(mod) 
p = parse(inspect.getsource(mod)) 

from ast import literal_eval 

vals = [] 
for node in p.body: 
    if isinstance(node, FunctionDef) and node.name == "my_method": 
     for node in walk(node): 
      if isinstance(node,Call) and node.func.id == "foobar": 
       vals.append([literal_eval(val) for val in node.args]) 

print(vals) 

[['apple', 'pear', 6], [1, 2, 3], ['a', 'b']] 

test.py的樣子：

def foobar(a=0, b=0, c=None): 
    return a, b, c 

def other_method(x,y,z): 
    return x,y,z 

def my_method(): 
    print('hello') 
    foobar('apple', 'pear', 6) 
    print('world') 
    foobar(1, 2, 3) 
    for i in range(10): 
     if i > 9: 
      s = foobar(4, 5, 6) 
      print(s) 
    return foobar('a', 'b') 


def my_method2(): 
    foobar('orange', 'tomatoe', 6) 
    foobar(10, 20, 30) 
    for i in range(10): 
     if i > 9: 
      foobar(40, 50, 60) 
    other_method("foo","bar","foobar") 
    return foobar('c', 'd') 

如果你有一個您需要將print('world')改爲foobar(a=1, b=2, c=3)

vals = [] 
for node in p.body: 
    if isinstance(node, FunctionDef) and node.name == "my_method": 
     for node in walk(node): 
      if isinstance(node, Call) and node.func.id == "foobar": 
       kws = node.keywords 
       if kws: 
        print("Found keywords",[(kw.arg, literal_eval(kw.value)) for kw in kws]) 
       else: 
        print([literal_eval(val) for val in node.args]) 

輸出：

[['apple', 'pear', 6], [], ['a', 'b'], [4, 5, 6]] 
['apple', 'pear', 6] 
('Found keywords', [('a', 1), ('b', 2), ('c', 3)]) 
['a', 'b'] 
[4, 5, 6] 

使用ast.Nodevisitor找到所有通話對象將返回所有調用所有功能 「FOOBAR」：

from ast import parse, NodeVisitor, literal_eval 
import importlib 
import inspect 

mod = "test" 
mod = importlib.import_module(mod) 
p = parse(inspect.getsource(mod)) 


class FindCall(NodeVisitor): 
    def __init__(self, *args): 
     if len(args) < 1: 
      raise ValueError("Must supply at least ine target function") 
     self.result = {arg: []for arg in args} 

    def visit_Call(self, node): 
     if node.func.id in self.result: 
      self.result[node.func.id].append(map(literal_eval, node.args)) 
     # visit the children 
     self.generic_visit(node) 



fc = FindCall("foobar") 
fc.visit(p) 
print(fc.result) 

輸出：

from pprint import pprint as pp 
pp(fc.result) 

{'foobar': [['apple', 'pear', 6], 
      [1, 2, 3], 
      [4, 5, 6], 
      ['a', 'b'], 
      ['orange', 'tomatoe', 6], 
      [10, 20, 30], 
      [40, 50, 60], 
      ['c', 'd']], 
'other_method': [['foo', 'bar', 'foobar']]} 

您可以再次添加對kwargs的搜索並僅搜索特定功能。