如何找出所有如下这种情况？

Question

如何查找python代码中所有未使用到的from与import的类库，例：from a import b# 下面的操作没用到b

Answer 1

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

比如这段代码:

from a import bb

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

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

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

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

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

import parserst = 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 bb = 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

Answer 2

楼主想要干的事情，在程序静态分析中，被称为DCE（无用代码的删除）。自己实现一个也不是不可以，不过如果急用的话就不太合适了。

现成的插件/库也是有很多的。我用过一段时间的pyflakes，它是一个代码检查插件，不仅可以满足楼主的要求，还有更多的功能。以楼主的代码为例，

$ echo "from a import b" > test.py$ pyflakes test.pytest.py:1: 'b' imported but unused

大概就是这样。