Python 上下文管理器与 with 语句从入门到精通作为一名从Python转向Rust的后端开发者我深刻体会到Python上下文管理器的强大和优雅。上下文管理器不仅可以帮助我们管理资源还可以使代码更加简洁、安全这让我在编写需要资源管理的代码时更加自信。今天我想分享一下Python上下文管理器与with语句的高级应用希望能帮助大家更好地理解和使用这个强大的特性。一、上下文管理器的基本概念1. 什么是上下文管理器上下文管理器是实现了__enter__和__exit__方法的对象它可以在进入和退出代码块时执行特定的操作。2. with 语句的基本用法我们可以使用with语句来使用上下文管理器它会在进入代码块时调用__enter__方法在退出代码块时调用__exit__方法。with open(file.txt, r) as f: content f.read() print(content) # 文件会在with语句结束后自动关闭二、高级应用技巧1. 自定义上下文管理器我们可以通过实现__enter__和__exit__方法来创建自定义的上下文管理器。class Timer: def __enter__(self): import time self.start time.time() return self def __exit__(self, exc_type, exc_val, exc_tb): import time self.end time.time() print(fElapsed time: {self.end - self.start} seconds) with Timer(): # 执行一些耗时的操作 import time time.sleep(1) # 输出: Elapsed time: 1.001234 seconds2. 使用 contextmanager 装饰器我们可以使用contextlib模块中的contextmanager装饰器来创建上下文管理器这样可以更简洁地实现。from contextlib import contextmanager contextmanager def timer(): import time start time.time() yield end time.time() print(fElapsed time: {end - start} seconds) with timer(): # 执行一些耗时的操作 import time time.sleep(1) # 输出: Elapsed time: 1.001234 seconds3. 嵌套使用上下文管理器我们可以嵌套使用多个上下文管理器这样可以管理多个资源。with open(input.txt, r) as f1, open(output.txt, w) as f2: content f1.read() f2.write(content) # 两个文件都会在with语句结束后自动关闭三、实用示例1. 管理数据库连接我们可以使用上下文管理器来管理数据库连接确保连接在使用后被正确关闭。import sqlite3 class DatabaseConnection: def __init__(self, db_path): self.db_path db_path self.conn None def __enter__(self): self.conn sqlite3.connect(self.db_path) return self.conn def __exit__(self, exc_type, exc_val, exc_tb): if self.conn: self.conn.close() with DatabaseConnection(example.db) as conn: cursor conn.cursor() cursor.execute(CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY, name TEXT)) cursor.execute(INSERT INTO users (name) VALUES (?), (Alice,)) conn.commit() # 数据库连接会在with语句结束后自动关闭2. 管理锁我们可以使用上下文管理器来管理锁确保锁在使用后被正确释放。import threading class LockManager: def __init__(self, lock): self.lock lock def __enter__(self): self.lock.acquire() return self def __exit__(self, exc_type, exc_val, exc_tb): self.lock.release() lock threading.Lock() with LockManager(lock): # 执行需要加锁的操作 print(Critical section) # 锁会在with语句结束后自动释放3. 临时修改环境变量我们可以使用上下文管理器来临时修改环境变量确保在退出时恢复原来的值。import os from contextlib import contextmanager contextmanager def temporary_env_var(key, value): original_value os.environ.get(key) os.environ[key] value yield if original_value is None: del os.environ[key] else: os.environ[key] original_value print(fOriginal value: {os.environ.get(TEST_VAR)}) with temporary_env_var(TEST_VAR, test_value): print(fTemporary value: {os.environ.get(TEST_VAR)}) print(fRestored value: {os.environ.get(TEST_VAR)})四、高级上下文管理器技术1. 处理异常上下文管理器的__exit__方法可以处理异常它接收三个参数exc_type异常类型、exc_val异常值和exc_tb异常回溯。class ErrorHandler: def __enter__(self): return self def __exit__(self, exc_type, exc_val, exc_tb): if exc_type: print(fError occurred: {exc_val}) # 返回True表示异常已经被处理 return True return False with ErrorHandler(): raise ValueError(Something went wrong) print(Continuing after error) # 输出: # Error occurred: Something went wrong # Continuing after error2. 返回值上下文管理器的__enter__方法可以返回一个值这个值会被赋值给with语句中的变量。class Resource: def __enter__(self): print(Acquiring resource) return Resource value def __exit__(self, exc_type, exc_val, exc_tb): print(Releasing resource) with Resource() as value: print(fUsing resource: {value}) # 输出: # Acquiring resource # Using resource: Resource value # Releasing resource3. 上下文管理器作为函数参数我们可以将上下文管理器作为函数参数这样可以更灵活地使用上下文管理器。from contextlib import contextmanager contextmanager def log_scope(scope): print(fEntering {scope}) yield print(fExiting {scope}) def process_data(data, context): with context: print(fProcessing data: {data}) process_data(test, log_scope(process)) # 输出: # Entering process # Processing data: test # Exiting process五、实战应用1. 事务管理我们可以使用上下文管理器来管理数据库事务确保事务在结束时被正确提交或回滚。import sqlite3 class Transaction: def __init__(self, conn): self.conn conn def __enter__(self): return self def __exit__(self, exc_type, exc_val, exc_tb): if exc_type: self.conn.rollback() print(Transaction rolled back) else: self.conn.commit() print(Transaction committed) conn sqlite3.connect(example.db) # 成功的事务 with Transaction(conn): cursor conn.cursor() cursor.execute(INSERT INTO users (name) VALUES (?), (Bob,)) # 失败的事务 with Transaction(conn): cursor conn.cursor() cursor.execute(INSERT INTO users (name) VALUES (?), (Charlie,)) raise ValueError(Intentional error) conn.close()2. 临时更改工作目录我们可以使用上下文管理器来临时更改工作目录确保在退出时恢复原来的目录。import os from contextlib import contextmanager contextmanager def temporary_cwd(path): original_cwd os.getcwd() os.chdir(path) yield os.chdir(original_cwd) print(fOriginal cwd: {os.getcwd()}) with temporary_cwd(/tmp): print(fTemporary cwd: {os.getcwd()}) print(fRestored cwd: {os.getcwd()})3. 资源池管理我们可以使用上下文管理器来管理资源池确保资源在使用后被正确归还到池中。from contextlib import contextmanager class ResourcePool: def __init__(self, size): self.resources [fResource {i} for i in range(size)] self.lock __import__(threading).Lock() contextmanager def acquire(self): with self.lock: if not self.resources: raise ValueError(No resources available) resource self.resources.pop() try: yield resource finally: with self.lock: self.resources.append(resource) pool ResourcePool(2) with pool.acquire() as resource1: print(fUsing {resource1}) with pool.acquire() as resource2: print(fUsing {resource2}) # 尝试获取第三个资源会失败 # with pool.acquire() as resource3: # print(fUsing {resource3}) print(fResources in pool: {pool.resources})六、总结Python的上下文管理器与with语句是一个非常强大的特性它可以帮助我们管理资源、处理异常、临时修改状态等。通过掌握自定义上下文管理器、使用contextmanager装饰器、嵌套使用上下文管理器等高级技巧我们可以编写更加简洁、安全、可维护的代码。作为一名从Python转向Rust的开发者我发现Python的上下文管理器与Rust的Drop特质有一些相似之处它们都可以确保资源在不再需要时被正确释放。但Python的上下文管理器更加灵活而Rust的Drop特质更加类型安全。这两种风格各有优缺点我们可以根据具体的场景选择合适的语言和技术。希望这篇文章能对你有所帮助如果你有任何问题或建议欢迎在评论区留言。