__exit__()方法的三个参数

• exc_type: 异常类型（如ValueError）
• exc_value: 异常实例对象
• traceback: 异常堆栈跟踪对象

如果__exit__()返回True，表示异常已被处理；返回False则异常会继续传播。

class DatabaseConnection:
    def __init__(self, db_name):
        self.db_name = db_name
        self.connection = None
        
    def __enter__(self):
        print(f"连接数据库: {self.db_name}")
        # 模拟可能发生的异常
        if "invalid" in self.db_name:
            raise ConnectionError("无效的数据库名称")
        self.connection = f"Connection to {self.db_name}"
        return self
    
    def __exit__(self, exc_type, exc_value, traceback):
        print("关闭数据库连接")
        # 清理资源
        self.connection = None
        
        # 处理特定异常
        if exc_type is ConnectionError:
            print(f"处理连接错误: {exc_value}")
            return True  # 异常已处理
        
        # 处理其他异常
        if exc_type is not None:
            print(f"发生异常: {exc_type.__name__}: {exc_value}")
            # 返回False将传播异常
            
    def execute_query(self, query):
        print(f"执行查询: {query}")
        # 模拟查询时可能发生的异常
        if "DROP" in query.upper():
            raise ValueError("不允许执行DROP操作")
        return "查询结果"

# 使用上下文管理器
try:
    with DatabaseConnection("example_db") as db:
        result = db.execute_query("SELECT * FROM users")
        print(result)
        # 模拟异常
        # db.execute_query("DROP TABLE users")
except Exception as e:
    print(f"外部捕获异常: {e}")
else:
    print("操作成功完成")

print("\n测试异常处理:")
try:
    with DatabaseConnection("invalid_db") as db:
        pass
except Exception as e:
    print(f"外部捕获异常: {e}")

代码解析：

• __enter__方法初始化资源，如果数据库名称无效会引发ConnectionError
• __exit__方法处理所有清理工作，无论是否发生异常
• __exit__方法专门处理ConnectionError异常并返回True
• 其他异常会被记录但继续传播
• execute_query方法模拟可能引发ValueError的操作

from contextlib import contextmanager

@contextmanager
def file_handler(filename, mode):
    try:
        file = open(filename, mode)
        print(f"打开文件: {filename}")
        yield file
    except Exception as e:
        print(f"处理文件异常: {e}")
        raise
    finally:
        print("关闭文件")
        file.close()

# 使用生成器实现的上下文管理器
try:
    with file_handler("example.txt", "w") as f:
        f.write("Hello, Context Manager!")
        # 模拟异常
        # raise IOError("模拟写入错误")
except Exception as e:
    print(f"捕获外部异常: {e}")

优点：

• 使用生成器函数替代类定义
• 代码更简洁，逻辑更清晰
• 使用try-finally确保资源释放
• 可以捕获并处理特定异常