如何在 python 脚本中找到所有 used 的函数

发布于07月24日

假设我有一个这样的python脚本

import numpy as np
x = [1, 2, 5, 3, 9, 6]
x.sort()
print(np.sum(x))

我想提取这个脚本中使用的所有函数.对于这个例子，结果应该是list.sort, np.sum, print.我可以列出所有内置函数，然后用dir(numpy)展开该列表.然后判断这些函数是否都出现在代码中.

但是如果有一个像np.linalg.norm(x)这样的函数呢？由于dir(numpy)不会返回norm，我将如何找到该函数.

注意:我知道可以调用代码中没有的许多其他函数.但我只想要代码中出现的那些函数.简单地说，我希望所有的文字/令牌将被vscode在黑暗+主题中涂成黄色.有没有可能用rope来解决这个问题？

推荐答案

你可以用ast module来做这样的事情:

import ast


code = """
import numpy as np
x = [1, 2, 5, 3, 9, 6]
x.sort()
print(np.sum(x))
"""

module = ast.parse(code)

for node in ast.walk(module):
    if isinstance(node, ast.Call):
        try:
            parent_name = node.func.value.id + '.'
        except AttributeError:
            parent_name = ''
        if isinstance(node.func, ast.Name):
            print(f'{parent_name}{node.func.id}')
        else:
            print(f'{parent_name}{node.func.attr}')

# output:
#  x.sort
#  print
#  np.sum

当然，有x.sort而不是list.sort，但我没有找到如何获得该方法的实际父类型.