Python with 详解：从入门到精通

共计 2718 个字符，预计需要花费 7 分钟才能阅读完成。

在 Python 中，with 语句被称为 上下文管理器（Context Manager），用于简化资源管理，确保资源在使用后被正确释放。它的主要作用是 自动管理资源的获取和释放，比如文件操作、数据库连接、多线程锁等。

本文将详细解析 with 语句的工作原理、应用场景，并通过多个示例帮助你彻底掌握 with 语句的使用。

1. with 语句的基本用法

with 语句通常用于资源管理，例如文件操作，确保文件在使用完毕后会自动关闭。

示例：使用 with 读取文件

# 传统方式打开文件（不推荐）file = open("example.txt", "r")
content = file.read()
file.close()  # 需要手动关闭文件
print(content)

如果程序在 file.read() 过程中抛出异常，file.close() 可能不会被执行，导致文件资源泄露。

使用 with 语句优化：

# 使用 `with` 语句
with open("example.txt", "r") as file:
    content = file.read()
print(content)  # 文件自动关闭

📌 优点：

• with 语句自动调用 file.close()，即使发生异常也不会导致文件未关闭。
• 代码更简洁，无需手动 close()。

2. with 语句的底层原理

with 语句的本质是使用 上下文管理器（Context Manager），其核心是 __enter__() 和 __exit__() 方法：

• __enter__()：在 with 代码块开始时调用，返回资源对象。
• __exit__()：在 with 代码块结束时自动执行，清理资源。

🔹 示例：自定义上下文管理器

class MyContext:
    def __enter__(self):
        print("进入 with 代码块")
        return self  # 可以返回需要的对象

    def __exit__(self, exc_type, exc_value, traceback):
        print("退出 with 代码块")
        # 处理异常（如果有）if exc_type:
            print(f"捕获异常: {exc_value}")
        return True  # 返回 True 表示异常已处理，不会向上抛出

with MyContext():
    print("执行代码...")
    raise ValueError("发生错误")  # 即使出错，`__exit__()` 仍会执行

输出：

进入 with 代码块
执行代码...
捕获异常: 发生错误
退出 with 代码块

可以看到：

• with 语句开始时，调用 __enter__()。
• with 代码块执行。
• 无论代码是否异常，都会执行 __exit__() 进行资源清理。

3. with 在文件操作中的应用

3.1 读取文件

with open("test.txt", "r") as file:
    content = file.read()
print(content)  # `with` 语句结束后文件已关闭

3.2 写入文件

with open("test.txt", "w") as file:
    file.write("Hello, Python!")

📌 自动关闭文件，避免 file.close() 遗漏。

3.3 逐行读取

with open("test.txt", "r") as file:
    for line in file:
        print(line.strip())  # 去除换行符

4. with 语句在其他场景的应用

4.1 线程锁

在多线程编程中，with 语句可以简化 threading.Lock() 的使用：

import threading

lock = threading.Lock()

def critical_section():
    with lock:
        print(f"线程 {threading.current_thread().name} 正在执行关键代码...")
        # 关键代码

📌 确保锁在代码块结束后自动释放，防止死锁。

4.2 数据库连接

使用 sqlite3 连接数据库时：

import sqlite3

with sqlite3.connect("database.db") as conn:
    cursor = conn.cursor()
    cursor.execute("CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY, name TEXT)")
    conn.commit()  # `with` 结束后自动提交

📌 with 语句确保数据库连接正确关闭，即使发生异常也不会影响数据库状态。

4.3 网络请求

在 requests 库中，with 语句用于管理 HTTP 连接：

import requests

with requests.get("https://www.example.com") as response:
    print(response.status_code)

📌 自动关闭网络连接，避免资源泄漏。

5. contextlib 简化上下文管理

Python 的 contextlib 提供了 contextmanager 装饰器，让我们可以更方便地创建上下文管理器。

示例：使用 @contextmanager 创建上下文管理器

from contextlib import contextmanager

@contextmanager
def my_resource():
    print("资源分配")
    yield "资源对象"
    print("资源释放")

with my_resource() as res:
    print(f"使用 {res}")

输出：

资源分配
使用 资源对象
资源释放

📌 yield 之前的代码相当于 __enter__()，yield 之后的代码相当于 __exit__()。

6. with 语句的常见错误

6.1 变量作用域

with open("test.txt", "r") as file:
    content = file.read()
print(file.read())  # ❌ 错误，文件已关闭

📌 解决方案：应在 with 语句内读取数据。

6.2 多个资源管理

如果需要同时管理多个资源，可以使用 多个 with 语句：

with open("input.txt") as input_file, open("output.txt", "w") as output_file:
    output_file.write(input_file.read())

📌 代码更简洁，自动管理多个资源。

7. 结论

🔹 适用场景：

• 文件操作（open()）
• 数据库连接（sqlite3、SQLAlchemy）
• 多线程锁（threading.Lock()）
• 网络请求（requests.get()）
• 自定义资源管理

你掌握 with 语句了吗？欢迎留言交流！🚀