python代码ast基础

AST（Abstract Syntax Tree）是一种抽象语法树，它将源代码转换为一种树形结构，用于表示程序的语法结构。在Python中，AST库允许你将源代码转换为AST，然后遍历和操作这个树形结构。这使得你能够更好地理解和维护代码。

Python的AST库提供了以下功能：

1. 将源代码转换为AST：使用ast.parse函数将源代码转换为AST。这个函数接受一个字符串参数，即要转换的源代码。

import ast

source_code = """
def add(a, b):
    return a + b

result = add(1, 2)
print(result)
"""

tree = ast.parse(source_code)

2. 遍历AST：使用ast.walk函数遍历AST。这个函数接受一个AST节点作为参数，并返回一个包含所有子节点的列表。

import ast

tree = ast.parse(source_code)

for node in ast.walk(tree):
    print(node)

3. 操作AST：使用各种ast模块中的函数来操作AST。例如，你可以使用ast.Assign来创建一个赋值语句，或者使用ast.Call来创建一个函数调用。

import ast

def add(a, b):
    return a + b

tree = ast.parse(source_code)

# 创建一个赋值语句
assign = ast.Assign(targets=[ast.Name(id='result', ctx=ast.Store())], value=ast.Call(func=ast.Name(id='add', ctx=ast.Load()), args=[ast.Num(n=1), ast.Num(n=2)], keywords=[]))

# 将赋值语句添加到AST中
tree.body.append(assign)

# 遍历AST并打印结果
for node in ast.walk(tree):
    print(node)

4. 编译和执行AST：使用compile函数将AST编译为字节码，然后使用exec或eval函数执行字节码。

import ast

tree = ast.parse(source_code)

# 编译AST为字节码
code = compile(tree, '<string>', 'exec')

# 执行字节码
exec(code)

总之，Python的AST库提供了一种强大的方法来处理和操作源代码，使得你能够更好地理解和维护代码。