例:
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
大概就是这样。