Python上下文管理器中yield导致__exit__未执行的原因及异常处理咨询

一、为什么__exit__里用yield会导致代码不执行？ #

Python的上下文管理器协议要求__exit__必须是普通函数，它负责在with块结束后执行清理逻辑，返回的布尔值决定是否压制with块中抛出的异常。

当你在__exit__里写yield self.connection.close()时，这个函数直接变成了生成器函数——调用它不会立即执行函数体代码，只会返回一个生成器对象。而Python处理上下文管理器时，只会把__exit__当作普通函数调用，不会去迭代这个生成器，所以函数里的打印语句、连接关闭逻辑根本不会被触发。

换成直接调用self.connection.close()，函数是普通函数，会按顺序执行所有代码；用return self.connection.close()时，哪怕close()返回None，函数依然是普通函数，代码会正常走完，只是返回值为None（相当于不压制异常）。

二、如何避免连接异常和__exit__执行出错时传递异常？ #

分两种场景针对性处理：

1. 连接数据库时（__enter__方法）出现异常 #

如果不想让连接失败的异常向外扩散，直接在__enter__内部捕获处理：

def __enter__(self):
    try:
        self.connection = cx_Oracle.connect(你的连接参数)
        return self.connection
    except cx_Oracle.DatabaseError as e:
        # 这里可以加日志记录、资源清理逻辑
        print(f"数据库连接失败: {e}")
        # 不想传异常就不重新抛出，也可以返回None供上层判断
        return None

或者在使用上下文管理器时，外层套try-except：

try:
    with OracleConnected(...) as conn:
        # 数据库操作逻辑
except cx_Oracle.DatabaseError as e:
    print(f"处理连接/操作异常: {e}")

2. __exit__方法执行时出错（比如关闭连接失败） #

__exit__内部如果抛出异常，会覆盖with块中可能存在的异常，并且直接向外传递。要避免这种情况，必须在__exit__内部把清理逻辑包裹在try-except里：

def __exit__(self, exc_type, exc_val, exc_tb):
    try:
        # 先判断连接是否存在，避免连接失败时触发AttributeError
        if hasattr(self, 'connection') and self.connection is not None:
            self.connection.close()
            print("数据库连接已关闭")
    except Exception as e:
        # 记录错误，不要让异常扩散出去
        print(f"关闭连接失败: {e}")
    
    # 返回True会压制with块中的异常，返回False/None则让异常继续传递
    # 根据业务需求选择返回值，比如不想压制异常就返回False
    return False

这里的核心是：把__exit__里的清理操作（比如关闭连接）放在try块内，捕获所有可能的异常并内部处理，不让它向外抛出。同时通过__exit__的返回值，控制是否压制with块本身的异常。

内容的提问来源于stack exchange，提问作者Денис Торопов