Python的ast模块如何使用

Python的ast（Abstract Syntax Trees，抽象语法树）模块是一个内置模块，用于解析Python代码并生成语法树。可以使用ast模块来分析、转换和生成Python代码。

以下是使用ast模块的一些常用步骤：

1. 导入ast模块

import ast

2. 读取Python代码并生成语法树

code_str = "print('Hello World')"
ast_tree = ast.parse(code_str)

3. 遍历语法树

class Visitor(ast.NodeVisitor):
    def visit_Call(self, node):
        print("Function name:", node.func.id)
        print("Arguments:", [ast.dump(arg) for arg in node.args])
        self.generic_visit(node)

visitor = Visitor()
visitor.visit(ast_tree)

4. 获取特定类型的节点

def find_function_calls(node):
    function_calls = []
    for child in ast.iter_child_nodes(node):
        if isinstance(child, ast.Call):
            function_calls.append(child)
        else:
            function_calls.extend(find_function_calls(child))
    return function_calls

function_calls = find_function_calls(ast_tree)

5. 修改语法树

class Transformer(ast.NodeTransformer):
    def visit_Str(self, node):
        return ast.Str('Modified string')

transformer = Transformer()
new_ast_tree = transformer.visit(ast_tree)
new_code_str = compile(new_ast_tree, filename="<ast>", mode="exec")
exec(new_code_str)

以上是一些ast模块的基本用法，可以根据需要进行更高级的操作。