python绘制ast并存储

Python自带的ast模块可以用于生成AST（Abstract Syntax Tree，抽象语法树）对象，可以使用第三方库（如graphviz）将AST可视化并存储为图像文件。

以下是一个示例代码，可以生成一个函数的AST可视化图并存储为PNG图像文件：

import ast
import graphviz

# 定义一个函数
def func(a, b):
    c = a + b
    d = a - b
    return c * d

# 解析函数的AST
tree = ast.parse(inspect.getsource(func))

# 定义AST节点和边的样式
node_attr = dict(style='filled', shape='box', align='left', fontsize='12', ranksep='0.1')
edge_attr = dict(arrowsize='0.7', fontsize='10', color='gray')

# 生成AST的可视化图
dot = graphviz.Digraph()
dot.attr('node', **node_attr)
dot.attr('edge', **edge_attr)

def add_nodes_edges(node):
    # 添加节点
    dot.node(str(node), str(node.__class__.__name__))
    # 遍历节点的子节点
    for child in ast.iter_child_nodes(node):
        # 添加边
        dot.edge(str(node), str(child))
        # 递归添加子节点和边
        add_nodes_edges(child)

add_nodes_edges(tree)

# 渲染并保存图像
dot.render('ast', format='png')

该代码会生成一个名为`ast.png`的PNG图像文件，可以在代码所在的目录下找到。如果需要修改文件名或路径，可以修改`render()`函数的参数。