with的常见用法

在 Python 中，with 语句用于简化资源管理，例如文件的打开和关闭、锁的管理等，确保在代码执行完后，资源自动释放。以下是一些常见的 with 用法：

1. 文件操作

使用 with 打开文件，确保在操作完成后自动关闭文件。

1 2	with open('file.txt', 'r') as file: content = file.read() # 文件会在此代码块执行完后自动关闭

2. 数据库连接

使用 with 语句管理数据库连接，保证在执行完毕后自动提交或回滚事务并关闭连接。

import sqlite3
with sqlite3.connect('database.db') as conn:
    cursor = conn.cursor()
    cursor.execute('SELECT * FROM table')
    results = cursor.fetchall()  # conn会自动关闭

3. 多线程或多进程锁

with 可以用来自动释放锁，避免出现死锁。

from threading import Lock
lock = Lock()
with lock:
    # 在此代码块中持有锁，代码块结束后会自动释放锁
    pass

4. 自定义上下文管理器

可以使用 with 来简化自定义对象的初始化和清理操作。自定义上下文管理器需要定义 __enter__ 和 __exit__ 方法。

class MyContext:
    def __enter__(self):
        print("开始")
    def __exit__(self, exc_type, exc_value, traceback):
        print("结束")

with MyContext():
    print("处理中")
# 输出:
# 开始
# 处理中
# 结束

5. 异常处理中的 Suppress

使用 contextlib.suppress 忽略特定的异常。

from contextlib import suppress
with suppress(FileNotFoundError):
    # 试图打开不存在的文件，不会报错
    open('nonexistent_file.txt')

6. 临时文件

使用 with 管理 TemporaryFile，确保临时文件在退出代码块时自动删除。

from tempfile import TemporaryFile
with TemporaryFile('w+t') as temp_file:
    temp_file.write("临时数据")
    temp_file.seek(0)
    print(temp_file.read())  # 代码块结束后文件被自动删除

通过 with 语句，Python 提供了简洁、安全的资源管理方式，有效减少了错误发生的概率。