Python with 关键字

在 Python 编程中，资源管理是一个重要但容易被忽视的环节。 with 关键字为我们提供了一种优雅的方式来处理文件操作、数据库连接等需要明确释放资源的场景。

with 是 Python 中的一个关键字，用于上下文管理协议（Context Management Protocol）。它简化了资源管理代码，特别是那些需要明确释放或清理的资源（如文件、网络连接、数据库连接等）。

为什么需要 with 语句？#

传统资源管理的问题#

我们先看一个典型的文件操作示例：

实例#

1
file = open('example.txt', 'r')
2
try:
3
    content = file.read()
4
    # 处理文件内容
5
finally:
6
    file.close()

这种写法存在几个问题：

容易忘记关闭资源 ：如果没有 try-finally 块，可能会忘记调用 close()
代码冗长 ：简单的文件操作需要多行代码
异常处理复杂 ：需要手动处理可能出现的异常

with 语句的优势#

with 语句通过上下文管理协议（Context Management Protocol）解决了这些问题：

自动资源释放 ：确保资源在使用后被正确关闭
代码简洁 ：减少样板代码
异常安全 ：即使在代码块中发生异常，资源也会被正确释放
可读性强 ：明确标识资源的作用域

with 语句的基本语法#

基础用法#

with 语句的基本形式如下：

语法格式#

1
with expression [as variable]:
2
    # 代码块

expression 返回一个支持上下文管理协议的对象
as variable 是可选的，用于将表达式结果赋值给变量
代码块执行完毕后，自动调用清理方法

文件操作示例#

最常见的 with 语句应用是文件操作：

实例#

1
with open('example.txt', 'r') as file:
2
    content = file.read()
3
    print(content)
4
# 文件已自动关闭

这段代码等价于前面的 try-finally 实现，但更加简洁明了。

with 语句的工作原理#

上下文管理协议#

with 语句背后是 Python 的上下文管理协议，该协议要求对象实现两个方法：

__enter__() ：进入上下文时调用，返回值赋给 as 后的变量
__exit__() ：退出上下文时调用，处理清理工作

执行流程#

异常处理机制#

__exit__() 方法接收三个参数：

exc_type ：异常类型
exc_val ：异常值
exc_tb ：异常追踪信息

如果 __exit__() 返回 True ，则表示异常已被处理，不会继续传播；返回 False 或 None ，异常会继续向外传播。

实际应用场景#

1. 文件操作#

实例#

1
# 同时打开多个文件
2
with open('input.txt', 'r') as infile, open('output.txt', 'w') as outfile:
3
    content = infile.read()
4
    outfile.write(content.upper())

2. 数据库连接#

实例#

1
import sqlite3
2

3
with sqlite3.connect('database.db') as conn:
4
    cursor = conn.cursor()
5
    cursor.execute('SELECT * FROM users')
6
    results = cursor.fetchall()
7
# 连接自动关闭

3. 线程锁#

实例#

1
import threading
2

3
lock = threading.Lock()
4

5
with lock:
6
    # 临界区代码
7
    print("这段代码是线程安全的")

4. 临时修改系统状态#

实例#

1
import decimal
2

3
with decimal.localcontext() as ctx:
4
    ctx.prec = 42  # 临时设置高精度
5
    # 执行高精度计算
6
    # 精度恢复原设置

创建自定义的上下文管理器#

类实现方式#

我们可以通过实现 __enter__ 和 __exit__ 方法创建自定义的上下文管理器：

实例#

1
class Timer:
2
    def __enter__(self):
3
        import time
4
        self.start = time.time()
5
        return self
6

7
    def __exit__(self, exc_type, exc_val, exc_tb):
8
        import time
9
        self.end = time.time()
10
        print(f"耗时: {self.end - self.start:.2f}秒")
11
        return False
12

13
# 使用示例
14
with Timer() as t:
15
    # 执行一些耗时操作
16
    sum(range(1000000))

使用 contextlib 模块#

Python 的 contextlib 模块提供了更简单的方式来创建上下文管理器：

实例#

1
from contextlib import contextmanager
2

3
@contextmanager
4
def tag(name):
5
    print(f"<{name}>")
6
    yield
7
    print(f"</{name}>")
8

9
# 使用示例
10
with tag("h1"):
11
    print("这是一个标题")

输出：

1
<h1>
2
这是一个标题
3
</h1>

常见问题与最佳实践#

常见错误#

1、错误地认为 with 只能用于文件 ：

实例#

1
# 错误：认为只有文件需要with
2
conn = sqlite3.connect('db.sqlite')
3
# 应该使用with语句

2、忽略__exit__的返回值 ：

实例#

1
class MyContext:
2
    def __exit__(self, exc_type, exc_val, exc_tb):
3
        # 忘记返回True/False可能导致异常处理不符合预期
4
        pass

最佳实践#

优先使用 with 管理资源 ：对于文件、网络连接、锁等资源，总是优先考虑使用 with 语句
保持上下文简洁 ： with 块中的代码应该只包含与资源相关的操作
合理处理异常 ：在自定义上下文管理器中，根据需求决定是否抑制异常
利用多个上下文 ：Python 允许在单个 with 语句中管理多个资源

总结要点#

关键点	说明
自动资源管理	`with` 语句确保资源被正确释放
上下文协议	需要实现 `__enter__` 和 `__exit__` 方法
异常安全	即使代码块中出现异常，资源也会被释放
常见应用	文件操作、数据库连接、线程锁等
自定义实现	可以通过类或 `contextlib` 创建自定义上下文管理器

with 语句是 Python 中一项强大的特性，它不仅能简化代码，还能提高程序的健壮性。掌握 with 语句的使用和原理，将帮助你写出更专业、更可靠的 Python 代码。

练习题#

1. with 语句基本用法 创建文件 test_with.py，完成以下操作：

用 with 语句读取文件内容
用 with 语句写入文件

知识点： with open(文件, 模式) as 变量:

参考答案：

1
# 写入文件
2
with open('test.txt', 'w') as f:
3
    f.write("Hello, Python!")
4

5
# 读取文件
6
with open('test.txt', 'r') as f:
7
    content = f.read()
8
    print(content)

2. 同时操作多个文件 创建文件 test_multi.py，完成以下操作：

用 with 同时打开两个文件
从一个文件读取内容，写入另一个文件

知识点： with open(文件1) as f1, open(文件2) as f2:

参考答案：

1
# 先创建源文件
2
with open('source.txt', 'w') as f:
3
    f.write("这是源文件内容")
4

5
# 同时读取和写入
6
with open('source.txt', 'r') as infile, open('dest.txt', 'w') as outfile:
7
    content = infile.read()
8
    outfile.write(content.upper())
9
    print("复制完成")

3. 自定义上下文管理器 创建文件 test_custom.py，完成以下操作：

用类实现一个计时器上下文管理器
记录代码执行时间

知识点： class 类名: def enter: def exit:

参考答案：

1
class Timer:
2
    def __enter__(self):
3
        import time
4
        self.start = time.time()
5
        return self
6

7
    def __exit__(self, exc_type, exc_val, exc_tb):
8
        import time
9
        self.end = time.time()
10
        print(f"耗时: {self.end - self.start:.2f}秒")
11
        return False
12

13
# 使用示例
14
with Timer():
15
    total = sum(range(1000000))
16
    print(f"计算结果: {total}")

4. contextlib 模块 创建文件 test_contextlib.py，完成以下操作：

用 @contextmanager 装饰器创建上下文管理器
实现一个简单的标签生成器

知识点： from contextlib import contextmanager、@contextmanager

参考答案：

1
from contextlib import contextmanager
2

3
@contextmanager
4
def tag(name):
5
    print(f"<{name}>")
6
    yield
7
    print(f"</{name}>")
8

9
# 使用示例
10
with tag("h1"):
11
    print("这是一个标题")
12

13
with tag("p"):
14
    print("这是一个段落")

团子和蛋糕

为什么需要 with 语句？#

传统资源管理的问题#

实例#

with 语句的优势#

with 语句的基本语法#

基础用法#

语法格式#

文件操作示例#

实例#

with 语句的工作原理#

上下文管理协议#

执行流程#

异常处理机制#

实际应用场景#

1. 文件操作#

实例#

2. 数据库连接#

实例#

3. 线程锁#

实例#

4. 临时修改系统状态#

实例#

创建自定义的上下文管理器#

类实现方式#

实例#

使用 contextlib 模块#

实例#

常见问题与最佳实践#

常见错误#

实例#

实例#

最佳实践#

总结要点#

练习题#

评论区

音乐

团子和蛋糕

Python with 关键字

Python with 关键字

为什么需要 with 语句？#

传统资源管理的问题#

实例#

with 语句的优势#

with 语句的基本语法#

基础用法#

语法格式#

文件操作示例#

实例#

with 语句的工作原理#

上下文管理协议#

执行流程#

异常处理机制#

实际应用场景#

1. 文件操作#

实例#

2. 数据库连接#

实例#

3. 线程锁#

实例#

4. 临时修改系统状态#

实例#

创建自定义的上下文管理器#

类实现方式#

实例#

使用 contextlib 模块#

实例#

常见问题与最佳实践#

常见错误#

实例#

实例#

最佳实践#

总结要点#

练习题#

相关阅读

评论区

音乐