Python with语句

在这个例子中，DatabaseConnection 类模拟了一个数据库连接对象，具有 connect 和 close 方法。DatabaseContextManager 类则是一个上下文管理器，它使用 DatabaseConnection 来管理数据库连接的建立和关闭。

当使用 with 语句时，__enter__ 方法会被调用以建立连接，并且连接对象会被绑定到 with 语句的 as 子句中的变量上（在这个例子中是 connection）。然后，with 语句块中的代码会执行。如果发生异常，异常会被捕获并传递到 __exit__ 方法中。无论是否发生异常，__exit__ 方法都会被调用以关闭数据库连接。最后，如果异常没有被 __exit__ 方法抑制（即 __exit__ 没有返回 True），则异常会继续传播。

# 传统的文件操作方式
file = open('example.txt', 'r')
try:
    data = file.read()
finally:
    file.close()

# 使用 with 语句
with open('example.txt', 'r') as file:
    data = file.read()

自定义上下文管理器

你可以通过定义一个类，并实现 __enter__() 和 __exit__() 方法来创建自己的上下文管理器。

class MyContextManager:
    def __enter__(self):
        print("Entering the context")
        return self  # 通常返回一些资源对象

    def __exit__(self, exc_type, exc_val, exc_tb):
        print("Exiting the context")
        # 可以处理异常
        # 如果处理成功，应该返回 True，否则返回 False 或不返回
        return False

# 使用自定义的上下文管理器
with MyContextManager() as cm:
    print("Inside the context")

线程锁

with语句也可以用于线程锁，确保线程安全

import threading

lock = threading.Lock()

with lock:
    # 临界区代码
    print("This block is thread-safe")

数据库连接

虽然具体的数据库库可能提供了自己的上下文管理器，但你也可以自己实现。

import sqlite3

class DatabaseConnection:
    def __enter__(self):
        self.conn = sqlite3.connect('example.db')
        self.cursor = self.conn.cursor()
        return self.cursor

    def __exit__(self, exc_type, exc_val, exc_tb):
        self.conn.commit()
        self.conn.close()

# 使用数据库连接上下文管理器
with DatabaseConnection() as cursor:
    cursor.execute("SELECT * FROM my_table")
    rows = cursor.fetchall()
    for row in rows:
        print(row)

总结

with语句的主要优点包括：