python - 解析 .py 文件，读取 AST，修改它，然后写回修改后的源代码

Question

我想以编程方式编辑 python 源代码。基本上我想读取一个 .py 文件，生成 AST ，然后写回修改后的python源代码(即另一个.py文件)。

有一些方法可以使用标准的python模块来解析/编译python源代码，例如ast或 compiler .但是，我认为它们中的任何一个都不支持修改源代码的方式(例如删除此函数声明)，然后将修改后的 python 源代码写回。

更新:我想这样做的原因是我想写一个 Mutation testing library对于 python，主要是通过删除语句/表达式，重新运行测试并查看中断。

Answer 1

Pythoscope对它自动生成的测试用例执行此操作，2to3 也是如此。 python 2.6 的工具(它将 python 2.x 源转换为 python 3.x 源)。

这两个工具都使用 lib2to3库，它是 python 解析器/编译器机制的实现，当它从源 -> AST -> 源往返时，可以保留源中的注释。

rope project如果您想进行更多重构(如转换)，可能会满足您的需求。

ast模块是您的另一个选择，there's an older example of how to "unparse" syntax trees back into code (使用解析器模块)。但是 ast 模块在对代码进行 AST 转换然后转换为代码对象时更有用。

redbaron项目也可能很合适(ht Xavier Combelle)