如何查找python代码中所有未使用到的from与import的类库

Question

例：

from a import b

# 下面的操作没用到b

如何找出所有这种情况？

Answer 1

假定你的代码中都没有直接从locals()/globals()中获取b的引用，那这还是比较容易的，下面给你举个例子吧。

比如这段代码:

from a import b
b

你可以用python的compiler模块去解析它:

import compiler
compiler.parse('from a import b\nb')

得到的输出为这段代码的AST(abstract syntax tree)

Module(None, Stmt([From('a', [('b', None)], 0), Discard(Name('b'))]))

有了AST，你就可以很容易地去处理它了，为了更简单一点，可以这样：

import parser
st = parser.suite('from a import b\nb')
print parser.st2list(st)

输出是：

[257, [267, [268, [269, [281, [283, [1, "from"], [288, [1, "a"]], [1, "import"], [286, [284, [1, "b"]]]]]], [4, ""]]], [267, [268, [269, [270, [327, [304, [305, [306, [307, [308, [310, [311, [312, [313, [314, [315, [316, [317, [318, [1, "b"]]]]]]]]]]]]]]]]]], [4, ""]], [0, ""]]

这就是一棵用list构成的树了，你要做的事情就是先序遍历，遍历的过程中用一个dict保存所有被import的name，遇到引用name的时候，记录它是否出现过，最后输出dict中没有出现过的name即可。

另外，如果有这样的情况:

from a import b
b = 123 #后续对b的引用与import的b无关

你也可以特殊处理一下赋值操作，检查被赋值的name是否在dict中且没有被引用过，如果是，输出它，从dict里删除，然后继续。

EDIT
补充一下，有以下这些import语句需要考虑：
//from Python/Include/graminit.h

#define import_stmt 281
#define import_name 282
#define import_from 283
#define import_as_name 284
#define import_as_names 286