如何处理嵌套方法链中的异常并回滚中间方法的变更？

针对你给出的代码场景，当method3抛出CustomException时要撤销method2中self.a = 3的赋值，有几种实用的方案：

方案1：延迟赋值（最简洁，优先推荐） #

核心思路是先执行可能抛出异常的逻辑，确认无异常后再修改属性，这样异常发生时属性根本不会被改动：

def method1(self):
    try:
        self.method2()
    except CustomException:
        print("Error")

def method2(self):
    # 先执行可能抛异常的method3，确认没问题再赋值
    self.method3()
    self.a = 3  # 只有method3正常执行完才会走到这一步

def method3(self):
    raise CustomException

这个方案适合method3不依赖self.a新值的场景，刚好匹配你的代码逻辑。

方案2：备份原有值，异常时恢复 #

如果必须先赋值再执行method3，可以先备份self.a的原有状态，异常触发时恢复回去：

def method1(self):
    try:
        self.method2()
    except CustomException:
        print("Error")

def method2(self):
    # 备份self.a的原有值，处理属性原本不存在的情况
    original_a = getattr(self, 'a', None)
    try:
        self.a = 3
        self.method3()
    except CustomException:
        # 恢复原有状态：如果原来有a就赋值回去，没有就删除这个属性
        if original_a is not None:
            self.a = original_a
        else:
            del self.a
        raise  # 重新抛出异常，让上层method1的except捕获

def method3(self):
    raise CustomException

方案3：上下文管理器（适合复杂场景复用） #

如果需要频繁处理这类"临时修改属性、异常回滚"的逻辑，可以写一个通用的上下文管理器：

class TempAttr:
    def __init__(self, obj, attr_name, new_value):
        self.obj = obj
        self.attr_name = attr_name
        self.original_value = getattr(obj, attr_name, None)
        self.new_value = new_value

    def __enter__(self):
        setattr(self.obj, self.attr_name, self.new_value)
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        # 只要发生异常，就恢复属性原有状态
        if exc_type is not None:
            if self.original_value is not None:
                setattr(self.obj, self.attr_name, self.original_value)
            else:
                delattr(self.obj, self.attr_name)
        # 返回False，让异常继续向上传播
        return False

# 原方法修改如下
def method1(self):
    try:
        self.method2()
    except CustomException:
        print("Error")

def method2(self):
    with TempAttr(self, 'a', 3):
        self.method3()  # 块内抛异常时，上下文会自动回滚self.a

def method3(self):
    raise CustomException

这个管理器可以复用在其他需要临时修改属性的场景中，代码更整洁。

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