Python上下文管理器__enter__与__exit__抛出异常的处理方案咨询

我正在开发一个用于远程控制电源供应器的Python包，把控制逻辑封装在PowerSupply类中，通过串口通信。使用时需要先将设备切换到远程模式，用完后切回本地模式，期望的上下文管理器流程如下：

初始化阶段 #

1. 打开串口
2. 将设备切换至远程控制模式

清理阶段 #

1. 关闭电压输出
2. 将设备切换至本地控制模式
3. 关闭串口

但上述每个步骤都可能抛出异常，我写了一个极简版实现：

import sys
import serial

class EMode:
    REMOTE = ...
    LOCAL = ...

class Out:
    OFF = ...
    ON = ...

class PowerSupply :
    def __init__(self, port: str) :
        self._device = serial.Serial(port)

    def __enter__(self) :
        try :
            self.setMode(EMode.REMOTE)
        except Exception as outer :
            try :
                self.__exit__(*sys.exc_info()) # 无论如何都要清理
            except Exception as inner :
                pass
            finally :
                raise outer # 不能忽略初始化错误
        else :
            return self

    def __exit__(self, _, __, ___) :
        try :
            self.setOut(Out.OFF)
            self.setMode(EMode.LOCAL)
        except Exception as e :
            raise e # 不能忽略清理错误
        finally :
            self.close()

    def setMode(...) :
        pass # 可能抛出异常

    def setOut(...) :
        pass # 可能抛出异常

使用方式：

with PowerSupply("/dev/ttyUSB0") as ps :
    ps.setMaxVoltage(5000);
    ps.setMaxCurrent(1000);
    ps.setVoltage(1000);
    ps.setOut(Out.ON);
    # 执行业务操作

我发现这个方案有个问题：__exit__方法里抛出的异常会覆盖with代码块中抛出的异常。虽然我的场景里所有错误都做了内部日志，影响不大，但其他项目里这可能是个严重问题。想请教下，当初始化和清理阶段都可能抛出异常时，推荐的处理方案是什么？

针对上下文管理器中初始化和清理阶段都可能抛出异常的场景，推荐以下几种处理思路：

1. 利用__exit__返回值控制传播，同时处理多异常 #

Python的__exit__方法返回True会抑制with块中的异常，返回False则让异常继续向上传播。你当前直接在__exit__里raise会覆盖原异常，正确做法是区分with块是否有异常，再结合清理异常处理：

• 若使用Python 3.11及以上，推荐用ExceptionGroup同时抛出所有异常（业务异常+清理异常），让调用者完整感知错误：

调整后的__exit__示例：

def __exit__(self, exc_type, exc_val, exc_tb):
    cleanup_errors = []
    
    # 逐个执行清理步骤，捕获异常
    try:
        self.setOut(Out.OFF)
    except Exception as e:
        cleanup_errors.append(e)
    
    try:
        self.setMode(EMode.LOCAL)
    except Exception as e:
        cleanup_errors.append(e)
    
    try:
        self._device.close()
    except Exception as e:
        cleanup_errors.append(e)
    
    # 组合异常处理
    if exc_val is not None:
        if cleanup_errors:
            raise ExceptionGroup("上下文管理器执行错误", cleanup_errors) from exc_val
        else:
            return False  # 无清理错误，让业务异常继续传播
    else:
        if cleanup_errors:
            raise ExceptionGroup("清理阶段错误", cleanup_errors)
        return True

• 若需兼容Python 3.11以下版本，可优先保留业务异常，将清理异常记录到日志后返回False；或自定义复合异常类，封装所有异常后抛出。

2. 拆分初始化逻辑，不在__init__中打开资源 #

你当前在__init__中直接打开串口，会导致初始化失败时需要额外清理。建议把资源打开（串口）移到__enter__中，这样初始化失败时资源未打开，无需复杂清理：

调整后的初始化代码：

class PowerSupply:
    def __init__(self, port: str):
        self._port = port
        self._device = None  # 先不打开串口

    def __enter__(self):
        try:
            self._device = serial.Serial(self._port)
            self.setMode(EMode.REMOTE)
        except Exception as init_exc:
            # 初始化失败，若串口已打开则尝试关闭
            if self._device is not None:
                try:
                    self._device.close()
                except:
                    pass
            raise init_exc  # 抛出初始化异常
        return self

3. 明确异常处理优先级 #

根据业务场景选择优先级：

• 若业务逻辑异常更重要（比如with块操作失败需优先排查），记录清理异常后优先抛出业务异常。
• 若设备状态正确性更重要（比如必须确保切回本地模式），优先抛出清理异常，但需在日志中记录业务异常信息。

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