Question

我正在使用＆＃39; ast＆＃39; python中的模块创建一个抽象语法树。我希望能够编辑AST（我正在使用＆＃39; ast.NodeTransformer＆＃39;进行编辑），然后获取新树并将其写入新的python文件。根据网站＆＃34; http://greentreesnakes.readthedocs.org/en/latest/index.html＆＃34;没有使用第三方软件包就无法做到这一点。这是真的，还是我可以使用＆＃39; ast＆＃39;将AST写入新的python文件？模块？如果是这样，我该怎么办？好像＆＃39; ast＆＃39;会支持这个。

Answer 1

你需要一个名为codegen.py的第三方模块，但它本身只是使用bulitin AST机制，非常简单。从那里你可以使用内置的ast.NodeTransformer机制来转换AST节点。

import ast
import codegen

class Visitor(ast.NodeTransformer):

    def visit_Num(self, node):
        return ast.Num(42)

x = Visitor()
t = ast.parse('x + y + z + 3')
out = x.visit(t)
print codegen.to_source(out)
# x + y + z + 42

将ast模块中的抽象语法树转换为新的python文件。

1 个答案: