解析.py文件,读取AST,修改它,然后写回修改后的源代码

import ast

with open("example.py", "r") as file:
    source_code = file.read()

tree = ast.parse(source_code)

class ASTVisitor(ast.NodeVisitor):
    def visit(self, node):
        print(type(node).__name__)
        self.generic_visit(node)

visitor = ASTVisitor()
visitor.visit(tree)

class ASTModifier(ast.NodeTransformer):
    def visit_FunctionDef(self, node):
        node.name = "modified_" + node.name
        return self.generic_visit(node)

modified_tree = ASTModifier().visit(tree)

with open("modified_example.py", "w") as file:
    ast.unparse(modified_tree, file)