Python-AST语法树

Python 的 Abstract Syntax Tree (AST) 是 Python 语言的抽象语法树表示，它可以用来表示 Python 代码的结构。

在 Python 中，我们可以使用 ast 模块来解析和转换 Python 代码的 AST。

解析 Python 代码并创建 AST 的方法如下：

import ast
 
code = """
def hello_world():
    print("Hello, world!")
"""
 
# 使用 ast.parse 方法解析代码，并返回一个模块对象
tree = ast.parse(code)

上述代码将创建一个表示 Python 代码的 AST，然后我们可以使用 ast.dump 方法打印出 AST 的文本表示，或者使用 ast.Module 对象的属性来访问 AST 的不同部分。

print(ast.dump(tree))

如果我们想要修改 AST，我们可以使用 ast.NodeTransformer 类。例如，我们可以创建一个修改器，它将所有的 print 函数调用替换为 print("Modified!")。

class ModifyPrint(ast.NodeTransformer):
    def visit_Print(self, node):
        print("Found a print statement.")
        return ast.fix_missing_locations(ast.Print(values=[ast.Str(s='Modified!')], nl=node.nl))
 
tree = ModifyPrint().visit(tree)
print(ast.dump(tree))

最后，我们可以使用 compile 函数将修改后的 AST 编译回代码字符串，并执行或 eval 它。

code = compile(tree, '<string>', 'exec')
exec(code)

以上就是如何使用 Python 的 AST 进行代码的解析、转换和生成的基本步骤。