Python的ContextManager：优雅管理资源

"== '__main__': # 示例代码演示 with DataConn('test.db') as db_conn: cursor = db_conn.cursor() cursor.execute("CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY, name TEXT)") print("数据库操作完成") ``` 在这个例子中，`DataConn` 类实现了 `__enter__` 和 `__exit__` 方法，使其成为一个上下文管理器。当使用 `with` 语句创建 `DataConn` 对象时，`__enter__` 方法会被调用，建立与数据库的连接，并返回连接对象给 `db_conn`。一旦 `with` 代码块执行完毕，`__exit__` 方法会被调用，确保无论代码块中是否发生异常，数据库连接都会被正确关闭。 2.2 使用上下文管理器的优势 - 错误处理：如果在 `with` 代码块中发生异常，`__exit__` 方法仍然会被调用，这使得资源清理（如关闭文件或断开数据库连接）可以在异常情况下也能执行，避免了资源泄露。 - 可读性：`with` 语句将相关的代码组织在一起，提高了代码的可读性和理解性，因为它清楚地标识了资源的获取和释放。 - 自动化：无需显式调用 `close()` 或类似方法，Python会自动处理这些细节，降低了出错的可能性。 3. 上下文管理器的其他应用 上下文管理器不仅仅用于文件和数据库连接，还可以应用于网络连接、锁定、内存缓存等任何需要开始和结束操作的场景。Python 标准库中的 `contextlib` 模块提供了更多的工具，如 `contextmanager` 函数，可以方便地将普通函数转换为上下文管理器。 4. 总结 Python 的上下文管理器是编写整洁、高效代码的关键特性之一。通过实现 `__enter__` 和 `__exit__` 方法，开发者可以创建自己的上下文管理器，确保资源的正确管理和异常处理。使用 `with` 语句，可以使代码更易读、更安全，同时简化了资源生命周期的管理。在日常编程中，善用上下文管理器能够提升代码质量，降低维护成本。