如何查找python代码中所有未使用到的from与import的类库
PHP中文网
PHP中文网 2017-04-17 11:20:45
[Python讨论组]

例:

from a import b

# 下面的操作没用到b

如何找出所有这种情况?

PHP中文网
PHP中文网

认证高级PHP讲师

全部回复(3)
怪我咯

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

大概就是这样。

PHP中文网

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

热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板
关于我们 免责申明 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送
PHP中文网APP
随时随地碎片化学习
PHP中文网抖音号
发现有趣的

Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号