聊聊Python中的上下文管理器：with语句背后的秘密

用Python的人对with open()肯定不陌生。但你知道with语句背后是怎么工作的吗这篇文章用最短的篇幅讲清楚。上下文管理器是什么简单说上下文管理器就是实现了__enter__和__exit__两个方法的对象。pythonclass MyContext: def __enter__(self): print(进入上下文) return self def __exit__(self, exc_type, exc_val, exc_tb): print(退出上下文) return False # False表示不屏蔽异常 with MyContext() as ctx: print(执行代码) # 输出 # 进入上下文 # 执行代码 # 退出上下文为什么需要它最常见的场景是资源管理——文件、数据库连接、锁等用完后需要释放或关闭。python# 不用with手动管理 f open(file.txt, r) try: data f.read() finally: f.close() # 用with自动管理 with open(file.txt, r) as f: data f.read() # 文件自动关闭contextlib模块的便捷方式用contextmanager装饰器可以把一个函数变成上下文管理器pythonfrom contextlib import contextmanager contextmanager def timer(): import time start time.perf_counter() try: yield finally: elapsed time.perf_counter() - start print(f耗时: {elapsed:.3f}秒) with timer(): # 执行你想计时的代码 sum(range(1000000)) # 输出: 耗时: 0.023秒实际应用场景数据库事务pythoncontextmanager def transaction(conn): try: yield conn conn.commit() except Exception: conn.rollback() raise with transaction(db_conn) as conn: conn.execute(INSERT INTO users ...)临时切换目录pythoncontextmanager def chdir(path): import os old_cwd os.getcwd() os.chdir(path) try: yield finally: os.chdir(old_cwd) with chdir(/tmp): # 在/tmp目录下执行操作 os.listdir(.)小结上下文管理器的本质就是自动化的try/finally。它让资源管理变得优雅避免了忘记释放资源导致的泄露问题。如果你在写一个需要获取-使用-释放的组件考虑用上下文管理器封装一下会让调用者的代码干净很多。