AST 变换术：构建自定义代码检查与自动修复插件

利用 Python 的抽象语法树（AST）模块，开发一个能检测特定模式并自动生成修复代码的 lint 工具，理解编译器前端的核心概念。 · 难度：入门 · +10XP

AST 变换术：构建自定义代码检查与自动修复插件

本教程从 ast.parse 开始，教会你如何遍历和修改 Python 代码的语法树。你将实现一个检测不安全 eval 使用的 lint 规则，并通过 ast.NodeTransformer 自动将 eval('expr') 替换为 ast.literal_eval('expr')。进一步地，你将学会如何将修改后的 AST 重新编译为可执行对象，或通过 ast.unparse 输出修复后的源代码。

import ast
class EvalToLiteral(ast.NodeTransformer):
    def visit_Call(self, node):
        if isinstance(node.func, ast.Name) and node.func.id == 'eval':
            # 替换为 ast.literal_eval
            new_call = ast.Call(
                func=ast.Attribute(value=ast.Name(id='ast'), attr='literal_eval'),
                args=node.args,
                keywords=[]
            )
            return new_call
        return self.generic_visit(node)