python ast

import ast

# 示例代码：使用 Python AST 模块解析和操作抽象语法树

# 定义一个简单的 Python 代码字符串
code = """
def hello(name):
    print(f"Hello, {name}!")
"""

# 使用 ast.parse 将代码字符串解析为抽象语法树 (AST)
tree = ast.parse(code)

# 打印 AST 的结构
print(ast.dump(tree, indent=4))

# 遍历 AST 并查找所有函数定义
for node in ast.walk(tree):
    if isinstance(node, ast.FunctionDef):
        print(f"Found function: {node.name}")

# 修改 AST 中的函数名
class RenameFunction(ast.NodeTransformer):
    def visit_FunctionDef(self, node):
        # 将函数名从 'hello' 改为 'greet'
        if node.name == 'hello':
            node.name = 'greet'
        return self.generic_visit(node)

# 应用 NodeTransformer 修改 AST
modified_tree = RenameFunction().visit(tree)

# 将修改后的 AST 转换回 Python 代码字符串
modified_code = ast.unparse(modified_tree)
print("Modified code:")
print(modified_code)

解释说明：

1. 导入模块：我们首先导入了 ast 模块，该模块用于解析和操作 Python 的抽象语法树 (AST)。
2. 定义代码字符串：我们定义了一个包含简单函数定义的 Python 代码字符串。
3. 解析代码为 AST：使用 ast.parse 方法将代码字符串解析为 AST。
4. 打印 AST 结构：使用 ast.dump 方法打印 AST 的结构，以便查看其内部表示。
5. 遍历 AST：通过 ast.walk 方法遍历 AST，并查找所有函数定义。
6. 修改 AST：定义了一个自定义的 NodeTransformer 类 RenameFunction，用于修改 AST 中的函数名。
7. 应用修改：使用 RenameFunction 类的实例对 AST 进行修改。
8. 转换回代码：使用 ast.unparse 方法将修改后的 AST 转换回 Python 代码字符串，并打印出来。

这个示例展示了如何使用 Python 的 AST 模块来解析、遍历和修改 Python 代码。