Python中with的高级使用

先看一段有问题的程序

f = open('data.json', 'r')   
# 加载JSON数据    
data = json.load(f)  
# 注意：这里没有显示地关闭文件

在这段程序中，文件被显式地打开并赋值给变量f，但随后并没有在代码块中显示地关闭文件。虽然在这个简单的例子中，Python解释器在脚本执行完毕后可能会自动关闭文件（这取决于Python解释器的具体实现和操作系统的行为），但在更复杂或更长的程序中，忘记关闭文件是一个常见的错误，它可能导致资源泄露、数据损坏或其他问题。

用with修改程序

with open('data.json', 'r') as f:    
    # 加载JSON数据    
    data = json.load(f)

这段程序使用了with语句来打开文件。with语句的好处在于它会在代码块执行完毕后自动关闭文件，即使是由于异常而提前退出的。这种自动关闭文件的方式可以避免文件描述符泄露，是一种更加安全和推荐的做法。

虽然这两段程序在功能上相同，但使用with语句是更推荐的做法，因为它能够自动管理资源（如文件），减少出错的可能性。在编写需要打开文件的Python代码时，应该优先考虑使用with语句。

with语句的高级使用示例

Python中的with语句主要用于上下文管理协议（context management protocol），它允许对象在代码块执行前后自动执行某些操作，比如自动关闭文件、获取和释放资源等。虽然with语句最常见的用途是打开和关闭文件，但它也可以用于更高级的场景，比如线程锁、数据库连接等。

以下是一些with语句的高级使用示例：

1. 自定义上下文管理器

你可以通过实现__enter__和__exit__方法来创建自己的上下文管理器。这两个方法分别在进入和退出with代码块时被自动调用。

class MyContextManager:  
    def __enter__(self):  
        print("Entering context")  
        # 初始化资源、设置状态等  
        return self  # 通常返回self或相关的资源对象  
  
    def __exit__(self, exc_type, exc_val, exc_tb):  
        print("Exiting context")  
        # 清理资源、保存状态等  
        # 如果不需要处理异常，则返回False；如果处理了异常，则返回True  
        # 这里简单返回True，表示不处理异常  
        return True  
  
# 使用自定义上下文管理器  
with MyContextManager():  
    print("Inside the context")

2. 线程锁（threading.Lock）

threading.Lock是一个线程锁，可以用来保护共享资源不被多个线程同时访问。使用with语句可以自动地加锁和解锁。

import threading  
  
lock = threading.Lock()  
  
def safe_function():  
    with lock:  
        # 访问或修改共享资源  
        print("Safe function executing")  
  
# 假设这里有一些线程调用了safe_function

3. 数据库连接

虽然Python的数据库连接库（如SQLite、MySQLdb等）通常不提供内置的上下文管理器支持，但你可以通过封装这些库来创建自己的上下文管理器，用于管理数据库连接的打开和关闭。

import sqlite3  
  
class DatabaseConnection:  
    def __init__(self, db_path):  
        self.db_path = db_path  
  
    def __enter__(self):  
        self.conn = sqlite3.connect(self.db_path)  
        self.cursor = self.conn.cursor()  
        return self.cursor  
  
    def __exit__(self, exc_type, exc_val, exc_tb):  
        self.cursor.close()  
        self.conn.close()  
  
# 使用数据库连接上下文管理器  
with DatabaseConnection('example.db') as cursor:  
    cursor.execute("SELECT * FROM some_table")  
    rows = cursor.fetchall()  
    # 处理rows

注意：在实际应用中，数据库连接池（如SQLAlchemy的session）是管理数据库连接更常见和更推荐的方式，因为它们提供了更高级别的连接管理、事务控制和对象关系映射（ORM）功能。

4. 临时文件或目录

虽然Python的tempfile模块提供了创建临时文件和目录的功能，但它本身并不直接支持with语句。但是，你可以通过结合使用tempfile和上下文管理器来简化临时资源的管理。

import tempfile  
  
class TemporaryFile:  
    def __enter__(self):  
        self.file = tempfile.TemporaryFile()  
        return self.file  
  
    def __exit__(self, exc_type, exc_val, exc_tb):  
        self.file.close()  
  
# 使用TemporaryFile上下文管理器  
with TemporaryFile() as temp_file:  
    temp_file.write(b"Hello, temporary file!")  
    temp_file.seek(0)  
    print(temp_file.read())  
# 此时临时文件已被自动删除

在这个例子中，TemporaryFile的__exit__方法只调用了close()方法，并没有显式删除文件。实际上，TemporaryFile的close()方法会在文件关闭后自动删除文件，所以这里不需要额外处理。但是，对于tempfile.NamedTemporaryFile或手动创建的临时目录，你可能需要在__exit__方法中显式地删除它们。 

另外，你体会一下，它像不像装饰器的作用？