在 Python 中生成 AST

在 Python 的底层实现中已经包含了源码到 AST 到 CodeObject 的转换过程，实际上 Python 也提供了一组工具，帮助我们直接控制 AST，如果熟练掌握的话，可以实现一些很有意思的魔法。

从源码到 AST

Python 已经内置了 ast 模块，可以直接从源码生成 AST，另外还有一组工具可以对 AST 做一些调整。首先从最基本的开始，从源码获得 AST 对象。

ast.parse(source, filename='<unknown>', mode='exec', *, type_comments=False, feature_version=None)

主要参数：

• source，待编译代码，字符串；
• filename，运行时错误信息会被输出到这个文件；
• mode，如果是单行代码为 “eval”，多行代码则为 “exec”；

其返回值为 AST 对象。

AST 对象是一个树状结构，每一个 Node 可能会有多个子节点，通过 ast.dump 可以方便的查看 AST 的内部。

import ast
  
src='''
a = 1
b = 2
c = a + b
'''


ast_node = ast.parse(src, "msg.log", mode="exec")

print(ast.dump(ast_node))

这样就可以得到输出：

Module(body