python - 使用正则表达式查找函数调用

Question

我正在尝试检测调用链之间的函数。

例如，我可以使用

re.search("([\w_]+)\((|[\W\d\w\,]+)\)", line)

寻找

print(len("AA"))

但它与以下代码相当不兼容:

print(i + len("AA") + j + len("BBB"))

帮助我。

Answer 1

ast 模块可能会更好地满足您的需求:

import ast

a = ast.parse('print(i + len("AA") + j + len("BBB"))')
print ast.dump(a)

>>>
Module(body=[Print(dest=None, values=[BinOp(left=BinOp(left=BinOp(left=Name(id='i',
ctx=Load()), op=Add(), right=Call(func=Name(id='len', ctx=Load()), args=[Str(s='AA')], 
keywords=[], starargs=None, kwargs=None)), op=Add(), right=Name(id='j', ctx=Load())), 
op=Add(), right=Call(func=Name(id='len', ctx=Load()), args=[Str(s='BBB')], keywords=[], 
starargs=None, kwargs=None))], nl=True)])