Python try-except块抛出异常前还原修改 如何避免重复写还原代码

澄清finally执行时机的误区 #

你之前认为finally不适用的判断存在偏差：Python中finally代码块的执行时机早于异常向外抛出的时机，无论try块内是正常执行完成、还是抛出异常，finally的逻辑都会先执行完成，再处理后续的异常传递或正常返回逻辑，完全符合你要求的「抛出错误前先还原属性」的需求。
因此你可以直接用finally简化原有代码，彻底消除重复的还原逻辑：

amount = instance.amount
instance.amount = 0
try:
    validate_instance(instance)
finally:
    instance.amount = amount

复用性更高的通用方案 #

如果这类临时修改属性的场景使用频率较高，可以基于上下文管理器封装通用工具，代码更简洁、语义也更清晰：

from contextlib import contextmanager

@contextmanager
def temporary_modify_attr(obj, attr_name, temp_value):
    # 保存属性原始值
    original_value = getattr(obj, attr_name)
    # 设置临时属性值
    setattr(obj, attr_name, temp_value)
    try:
        yield
    finally:
        # 无论是否抛出异常都优先还原属性
        setattr(obj, attr_name, original_value)

使用时只需几行代码即可完成需求：

with temporary_modify_attr(instance, 'amount', 0):
    validate_instance(instance)

方案优势 #

• 完全消除重复的还原代码，避免漏写还原逻辑的低级错误
• 作用域清晰，仅在with块内生效，可读性强
• 通用工具可全局复用，适配所有对象属性临时修改的场景

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