有没有工具可以检查我从“疯狂”导入的模块中使用了哪些名称？

Question

为了我的研究，我一直在使用python进行计算。为了清理我糟糕的代码，我一直在阅读davidgoodger的Code Like a Pythonista: Idiomatic Python。在

在本文中，Goodger建议不要使用“通配符”导入表单

from module import *

我的代码使用了很多这样的代码。我想清理我的代码，但我不知道如何清理。我很好奇是否有一种方法可以检查我在代码中使用了module中的名称。这样我就可以显式地导入这些名称，或者用module.name替换这些名称。有没有专门设计的工具来完成这样的任务？在

Answer 1

这里有一个更简单的解决方案。它使用ast模块将代码从文件中剥离出来，然后将其与inspect模块找到的可用函数列表进行比较。只需在运行之前替换yourfilename和yourmodulename。在

import ast, inspect, yourmodulename as mymodule
filename='yourfilename.py'
tab = ' '*4

funcs = {m[0] for m in inspect.getmembers(mymodule) 
        if str(m[1])[1:].split(' ')[0] in ('function', 'class') and 
        inspect.getmodule(m[1]) == mymodule}
with open(filename) as f:
    code = ast.parse(f.read())
    words = {node.id for node in ast.walk(code) if isinstance(node, ast.Name)}
    print('from', mymodule.__name__, 'import (')
    out = tab
    for word in (', '.join((sorted(words & funcs)))+')').split():
        if len(out + word) > 80:
            print(out.rstrip())
            out = tab + word + ' '
        else:
            out += word + ' '
    if out:
        print(out.rstrip())

我的脚本和我的脚本检查了几十行。在

Edit：添加了检查，以确保它只列出实际属于所需模块一部分的函数，而不是它的子模块。另外，修改后使其同时列出要导入的类和函数。在