4-上下文管理器

4-1-概念

1. 上下文管理器是一种实现了 `__enter__()` 和 `__exit__()` 方法的对象；
2. 用于管理资源的生命周期，如文件的打开和关闭、数据库连接的建立和断开等。
3. 使用 `with` 语句可以方便地使用上下文管理器。

4-2-示例

4-2-1-代码

class FileHandler:
    def __init__(self, filename, mode):
        self.filename = filename
        self.mode = mode
        self.file = None

    def __enter__(self):
        self.file = open(self.filename, self.mode)
        return self.file

    def __exit__(self, exc_type, exc_value, traceback):
        if self.file:
            self.file.close()


with FileHandler('testPython.txt', 'w') as f:
    f.write('Hello, World!')

4-2-2-输出结果

4-2-3-代码分析

1. `FileHandler` 类实现了上下文管理器协议。
2. `__enter__()` 方法在进入 `with` 语句块时被调用，用于打开文件并返回文件对象；
3. `__exit__()` 方法在离开 `with` 语句块时被调用，用于关闭文件。

4-3-案例

Python 的上下文管理器（Context Manager）是一种用于管理资源的机制，它确保资源在使用完毕后被正确释放，避免资源泄漏。上下文管理器通常使用 `with` 语句来调用，其核心是通过 `__enter__` 和 `__exit__` 方法实现的。以下是一些常见的使用场景及对应的例子。

4-3-1-文件操作

在文件操作中，使用上下文管理器可以确保文件在使用完毕后被正确关闭，即使在操作过程中发生异常。

4-3-1-1-例子

# 使用上下文管理器打开文件
with open('example.txt', 'w') as file:
    file.write('Hello, World!')

# 不需要手动调用 file.close()，文件会在 with 块结束时自动关闭

4-3-1-2-输出结果

4-3-1-3-代码分析

1. `open` 函数返回的文件对象是一个上下文管理器。
2. `with` 语句会自动调用文件对象的 `__enter__` 方法打开文件，并将文件对象赋值给 `file` 变量。
3. 当 `with` 块结束时，无论是否发生异常，都会自动调用文件对象的 `__exit__` 方法关闭文件。

4-3-2-数据库连接

在数据库操作中，使用上下文管理器可以确保数据库连接在使用完毕后被正确关闭，避免资源泄漏。

4-3-2-1-例子

import sqlite3

# 自定义上下文管理器类
class DatabaseConnection:
    def __init__(self, db_name):
        self.db_name = db_name
        self.connection = None

    def __enter__(self):
        self.connection = sqlite3.connect(self.db_name)
        return self.connection

    def __exit__(self, exc_type, exc_val, exc_tb):
        if self.connection:
            self.connection.close()


# 使用上下文管理器连接数据库
with DatabaseConnection('example.sql') as conn:
    cursor = conn.cursor()
    cursor.execute('CREATE TABLE IF NOT EXISTS test (id INTEGER PRIMARY KEY, name TEXT)')
    cursor.execute('INSERT INTO test (name) VALUES (?)', ('John',))
    conn.commit()

4-3-2-2-输出结果

4-3-2-3-代码分析

1. `DatabaseConnection` 是一个自定义的上下文管理器类。
2. `__enter__` 方法负责建立数据库连接并返回连接对象。
3. `__exit__` 方法负责在 `with` 块结束时关闭数据库连接。

4-3-3-线程锁

在多线程编程中，使用上下文管理器可以确保线程锁在使用完毕后被正确释放，避免死锁。

4-3-3-1-例子

import threading

# 创建线程锁
lock = threading.Lock()

# 使用上下文管理器获取和释放锁
with lock:
    # 执行需要线程安全的操作
    print('线程锁已获取，执行操作...')
    # 模拟一些操作
    for i in range(3):
        print(i)
    print('操作完成，线程锁已释放。')

4-3-3-2-输出结果

4-3-3-3-代码分析

1. `threading.Lock()` 返回的锁对象是一个上下文管理器。
2. `with` 语句会自动调用锁对象的 `__enter__` 方法获取锁，并在 `with` 块结束时自动调用 `__exit__` 方法释放锁。

4-3-5-临时文件和目录

在需要创建临时文件或目录时，使用上下文管理器可以确保这些临时资源在使用完毕后被正确清理。

4-3-5-2-例子

import tempfile

# 使用上下文管理器创建临时文件
with tempfile.TemporaryFile('w+t') as temp_file:
    temp_file.write('This is a temporary file.')
    temp_file.seek(0)
    print(temp_file.read())

# 临时文件会在 with 块结束时自动删除

4-3-5-2-输出结果

4-3-5-3-代码分析

1. `tempfile.TemporaryFile()` 返回的临时文件对象是一个上下文管理器。
2. `with` 语句会自动处理临时文件的创建和删除，确保资源的正确管理。