大多数教程展示了平凡的 with open(...) 示例，然后就停止了。这里是我在实际代码中实际使用的三个模式，以及每个模式防止的故障模式。

案例1：资源清理与错误处理

当您有必须清理的资源时，无论发生什么：

 1class DatabaseConnection:
 2    def __init__(self, config):
 3        self.config = config
 4        self.connection = None
 5    
 6    def __enter__(self):
 7        self.connection = create_connection(self.config)
 8        return self.connection
 9    
10    def __exit__(self, exc_type, exc_val, exc_tb):
11        if self.connection:
12            self.connection.close()

为什么这很重要

  • 即使在异常情况下也会清理资源
  • 防止资源泄漏
  • 使代码更可靠

案例2：临时状态管理

当您需要临时更改某些状态，然后恢复它：

 1class TempDir:
 2    def __init__(self, base_path="/tmp"):
 3        self.base_path = base_path
 4        self.temp_path = None
 5    
 6    def __enter__(self):
 7        import tempfile
 8        self.temp_path = tempfile.mkdtemp(dir=self.base_path)
 9        return self.temp_path
10    
11    def __exit__(self, exc_type, exc_val, exc_tb):
12        import shutil
13        if self.temp_path and os.path.exists(self.temp_path):
14            shutil.rmtree(self.temp_path)

为什么这很重要

  • 自动清理临时文件
  • 防止磁盘空间浪费
  • 简化错误处理

案例3：性能监控和计时

当您需要测量代码执行时间：

 1import time
 2
 3class Timer:
 4    def __enter__(self):
 5        self.start = time.time()
 6        return self
 7    
 8    def __exit__(self, exc_type, exc_val, exc_tb):
 9        self.end = time.time()
10        print(f"执行时间: {self.end - self.start:.2f} 秒")

为什么这很重要

  • 自动计时代码块
  • 即使在异常情况下也会报告时间
  • 便于性能监控

关键要点

  1. 上下文管理器不仅仅用于文件 - 它们适用于任何需要设置/清理的资源
  2. 异常安全 - __exit__ 方法总是被调用，即使在异常情况下
  3. 可组合 - 您可以嵌套多个上下文管理器
  4. 可重用 - 创建一次，使用多次

实际应用

这些模式在实际代码中非常有用，特别是当您处理数据库连接、临时文件、网络连接或需要精确计时的操作时。

记住：上下文管理器是Python中资源管理的瑞士军刀。学习使用它们将使您的代码更健壮和更易维护。