笔记大全

python中的with怎么使用

更新时间：2023-10-16

概述

在 Python 中有许多用于文件操作的函数和方法，但是如果使用不当，则容易引起各种问题，例如文件没有关闭、异常处理等。因此，Python 提供了与上下文管理器（Context Manager）相关的 with 语句，以便更安全地使用文件（以及其他资源）。

基本语法

with 语句的基本语法如下：

with context_expression [as target(s)]:
    with_suite

这里的 context_expression 可以是支持上下文管理器协议的任何对象（例如文件、锁、数据库连接等）。

在 with 语句块中，我们可以执行对 context_expression 进行设置或清理的语句。

如果指定了 as target(s) 子句，则 context_expression.__enter__() 方法的返回值将分配给变量 target(s)。

文件操作示例

在通常的文件操作中，我们需要关闭打开的文件对象，以释放其资源。而使用 with 语句，可以确保文件在不需要时被正确地关闭。下面是一个示例：

with open('example.txt', 'r') as f:
    contents = f.read()
    # do something with contents
# 文件在此处已自动关闭

同时，如果在读取文件期间出现了错误，则文件将在程序转到 except 语句块之前自动关闭。

自定义类的上下文管理器

除了内置类型文件之外，我们还可以编写自定义类的上下文管理器，例如以下示例：

class CustomContextManager:
    def __init__(self, value):
        self.value = value
    def __enter__(self):
        print('Entering with context')
        return self.value
    def __exit__(self, exc_type, exc_val, exc_tb):
        print('Exiting with context')
        # return True # uncommenting this line would suppress any exceptions

# 使用自定义上下文管理器
with CustomContextManager(5) as value:
    print(value)
# 输出：
# Entering with context
# 5
# Exiting with context

总结

通过使用 with 语句，我们可以更安全、更方便地管理资源。with 语句使用起来非常简单，并且减少了程序员的工作负担，因为它维护并关闭文件、锁等对象。

对于更高级的场景，如果需要实现自定义的上下文管理器，只需定义 __enter__() 和 __exit__() 方法即可。

总之，with 语句是实现正确和安全的资源管理的一种强大方式，它可以用于几乎所有需要使用到上下文管理器的场景。

c语言编程笔录