使用ast和astor改写无类Python代码时遇AttributeError问题求助

错误原因 #

你在处理ast.ClassDef节点时错误地尝试访问node.args属性，但**ast.ClassDef根本没有args属性**——这个属性是ast.FunctionDef（函数定义节点）特有的，用来存储函数的参数列表。错误提示里的代码行args=[ast.arg(arg='self', annotation=None)] + node.args.args,，是你在处理类定义时误写的逻辑，把ClassDef当成了FunctionDef来操作。

修复方案与完整实现 #

要实现“将类替换为函数”的目标，核心逻辑应该是：遍历AST找到类定义，提取类中的方法并转换为独立函数（移除self参数），处理命名冲突，最后移除原类定义。以下是修正后的完整代码：

import ast
import astor

def rewrite_without_classes(code):
    tree = ast.parse(code)
    class_replacements = {}

    def visit_and_replace(node, class_name=None):
        # 处理类定义节点
        if isinstance(node, ast.ClassDef):
            class_name = node.name
            class_replacements[class_name] = {}
            # 遍历类体中的所有子节点（方法、变量等）
            for sub_node in node.body:
                visit_and_replace(sub_node, class_name)
            # 从AST中移除原类定义
            tree.body.remove(node)
        
        # 处理类中的方法节点
        elif isinstance(node, ast.FunctionDef) and class_name is not None:
            # 生成新的函数定义：移除self参数，给函数名加类前缀避免冲突
            new_args = ast.arguments(
                args=node.args.args[1:] if node.args.args else [],  # 跳过第一个self参数
                defaults=node.args.defaults,
                kw_defaults=node.args.kw_defaults,
                kwarg=node.args.kwarg,
                vararg=node.args.vararg
            )
            new_func = ast.FunctionDef(
                name=f"{class_name}_{node.name}",
                args=new_args,
                body=node.body,
                decorator_list=node.decorator_list
            )
            class_replacements[class_name][node.name] = new_func
            # 将转换后的函数添加到全局作用域
            tree.body.append(new_func)

    # 遍历AST顶层节点（用copy避免遍历中修改列表导致的问题）
    for stmt in tree.body.copy():
        visit_and_replace(stmt)

    # 生成修改后的代码
    return astor.to_source(tree)

# 测试示例
if __name__ == "__main__":
    test_code = """
class Calculator:
    def add(self, a, b):
        return a + b
    
    def multiply(self, x, y=10):
        return x * y
"""
    print(rewrite_without_classes(test_code))

关键修复点说明 #

• 节点类型区分：明确区分ClassDef和FunctionDef节点，对类节点只遍历其body属性获取内部方法，绝不访问args。
• 方法转函数的参数处理：保留原方法的默认参数、关键字参数等信息，仅移除第一个self参数（如果存在）。
• 命名冲突避免：给转换后的函数名添加原类名前缀（比如Calculator_add），防止不同类中的同名方法冲突。
• AST修改安全性：遍历顶层节点时使用tree.body.copy()，避免遍历过程中修改列表导致的遍历异常。

内容的提问来源于stack exchange，提问作者FrostDream