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

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

例:

from a import b

# 下面的操作没用到b

如何找出所有这种情况?


python有个代码检查工具,pyflake,还可以集成到vim,在vim中用cc检查


假定你的代码中都没有直接从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

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

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

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

大概就是这样。

【热门文章】
【热门文章】