Python上下文管理器：with语句与__enter__/__exit_

在Python编程中上下文管理器Context Manager是一个强大而优雅的特性。它通过with语句实现了资源的自动获取和释放让代码更加简洁、可读性更强同时有效避免了资源泄漏问题。本文将深入探讨上下文管理器的工作原理、实现方式以及实际应用场景。为什么需要上下文管理器在没有上下文管理器之前我们处理文件操作通常是这种方式# 传统方式 - 容易出错 f open(data.txt, r) data f.read() f.close() # 如果上面出错这里不会执行使用try-finally改进# try-finally 方式 - 代码冗长 f open(data.txt, r) try: data f.read() finally: f.close() # 确保关闭但代码繁琐而使用with语句# with 语句 - 简洁优雅 with open(data.txt, r) as f: data f.read() # 自动关闭无需手动处理⚙️ 上下文管理器的工作原理上下文管理器的核心是两个特殊方法__enter__(): 进入上下文时调用返回的资源会赋值给as后的变量__exit__(exc_type, exc_val, exc_tb): 退出上下文时调用处理异常清理工作️ 自定义上下文管理器方式一基于类的实现class DatabaseConnection: def __init__(self, host, port): self.host host self.port port self.connection None def __enter__(self): print(f 连接到数据库 {self.host}:{self.port}) self.connection fConnection({self.host}:{self.port}) return self.connection def __exit__(self, exc_type, exc_val, exc_tb): print(f 关闭数据库连接) self.connection None # 返回True表示异常已被处理不再传播 return False # 使用示例 with DatabaseConnection(localhost, 3306) as conn: print(f执行查询: {conn}) # 输出 # 连接到数据库 localhost:3306 # 执行查询: Connection(localhost:3306) # 关闭数据库连接方式二使用 contextlib 装饰器更简洁from contextlib import contextmanager contextmanager def managed_resource(name): print(f 获取资源: {name}) resource fResource({name}) try: yield resource # 这里是 with 语句 as 变量的值 finally: print(f️ 释放资源: {name}) # 使用示例 with managed_resource(file_handle) as res: print(f使用: {res}) # 输出 # 获取资源: file_handle # 使用: Resource(file_handle) # ️ 释放资源: file_handle 实际应用场景1. 数据库事务管理from contextlib import contextmanager contextmanager def transaction(conn): try: conn.begin() yield conn conn.commit() print(✅ 事务提交成功) except Exception as e: conn.rollback() print(f❌ 事务回滚: {e}) raise # 使用 with transaction(db_conn) as conn: conn.execute(INSERT INTO users VALUES (1, Alice)) conn.execute(INSERT INTO orders VALUES (1, 100))2. 性能计时器import time from contextlib import contextmanager contextmanager def timer(label): start time.time() try: yield finally: elapsed time.time() - start print(f⏱️ {label}: {elapsed:.4f}秒) # 使用 with timer(数据处理): result sum(range(1000000)) # 输出⏱️ 数据处理: 0.0234秒3. 临时修改环境import os from contextlib import contextmanager contextmanager def temporary_env(key, value): old_value os.environ.get(key) os.environ[key] value try: yield finally: if old_value is None: del os.environ[key] else: os.environ[key] old_value # 使用 with temporary_env(DEBUG, 1): print(fDEBUG {os.environ.get(DEBUG)}) # DEBUG 1 print(fDEBUG {os.environ.get(DEBUG)}) # 恢复原值嵌套上下文管理器Python支持同时管理多个上下文# 同时打开多个文件 with open(input.txt) as fin, open(output.txt, w) as fout: for line in fin: fout.write(line.upper()) # 等价于嵌套写法 with open(input.txt) as fin: with open(output.txt, w) as fout: for line in fin: fout.write(line.upper())

Python上下文管理器：with语句与enter/exit完全指南

相关推荐

SetDPI：3步掌握Windows命令行DPI调整的终极方案

Win11Debloat：3分钟免费优化Windows系统，让电脑重获新生

Unity Mod Manager终极指南：5步轻松管理Unity游戏模组

更多资讯

【数据结构】时间复杂度和空间复杂度介绍

数学建模高效学习：优秀论文精读与团队协作实战指南

USB设备识别Windows系统：枚举特征、描述符请求与固件实现

SystemVerilog function与task边界、选型与避坑

VxWorks 6.9虚拟机搭建指南：VMware上运行与Workbench调试完整教程

嵌入式固件烧录与OTA升级实战：从编译到远程更新的完整链路

今日更新

MATLAB+Yalmip+CPLEX实战：综合能源系统优化调度全流程解析

I3C比I2C快10倍？RK3576实战：速率、DTS配置与混合总线避坑指南

字符串转对象：JSON.parse、new Function与URLSearchParams

本周精选

如何彻底离线使用 Laya-CoreML：Hugging Face 下载、缓存机制与 local_files_only 详解

基于1Panel 部署MySQL MCP 服务：TaoToken 统一 Key 接入与 config.toml 配置实战

崇州园区营销网站建设避坑指南：5大注意事项保备案安全

本月热门

基于YOLO的焊缝缺陷检测：从数据标注到工业部署全流程实践

分布式服务的决策记录

创业产品的首版范围

读完文章，想聊聊您的网站？