本文链接：https://blog.csdn.net/PlutoZuo/article/details/134393156

Python之with

文章目录

Python之with

with简介

在Python中，with语句是一种处理资源的特殊方式，它可以确保在使用资源之后正确地释放它们这种机制被称为上下文管理协议有时，我们可能需要在完成某些操作后自动关闭文件或清理其他类型的资源使用with语句可以确保这些操作在程序执行完毕后得到执行

例如，当我们打开一个文件进行读取时，我们通常需要确保在完成读取后关闭文件以避免资源泄漏这可以通过使用with语句来实现：

with open('filename.txt', 'r') as file:
    data = file.read()

在这个例子中，open()函数返回的文件对象将作为with语句的上下文管理器在with语句的代码块中，我们可以像平常一样使用文件对象一旦代码块执行完毕，文件对象就会被自动关闭，无需显式调用close()方法

with语句不仅可以用于文件操作，还可以用于其他需要清理资源的场景例如，在使用线程或进程时，我们可以使用with语句来自动调用join()或joinall()方法来等待线程完成并释放资源

需要注意的是，为了能够正确地使用with语句，我们需要确保操作的资源实现了上下文管理协议，即它们需要定义__enter__()和__exit__()方法来管理资源的生命周期如果操作的资源不支持上下文管理协议，我们仍然可以使用传统的打开-使用-关闭模式来进行操作

使用with语句块可以避免忘记关闭文件而导致资源泄漏的问题，因为它会在代码块执行完毕后自动关闭文件，即使在代码块中发生了异常也是如此除了文件对象，with关键字还可以用于其他上下文管理器，例如线程锁、队列等，以确保在操作完成后正确地释放资源

with原理

Python中的with关键字用于上下文管理，它可以确保在代码块执行完毕后正确地清理资源，从而避免资源泄漏的错误它是由Python的内置函数__enter__和__exit__实现的，这两个方法分别用于进入和退出代码块

当使用with关键字时，Python会首先调用__enter__方法，将对象传入代码块中，并在代码块执行完毕后自动调用__exit__方法，清理资源

例如，在使用文件对象时，__enter__方法返回文件对象本身，__exit__方法用于关闭文件这样，在代码块中读取文件内容后，Python会自动关闭文件，避免了资源泄漏的问题

下面是一个示例代码，展示了如何使用with来打开文件并进行读写操作：

class File:
    def __init__(self, filename):
        self.filename = filename
        self.file = None
    
    def __enter__(self):
        self.file = open(self.filename, 'r')
        return self.file
    
    def __exit__(self, exc_type, exc_value, traceback):
        self.file.close()

在这个例子中，我们定义了一个上下文管理器File，它用于打开文件并返回文件对象在__enter__方法中，我们打开文件并返回文件对象本身在__exit__方法中，我们关闭文件这样可以确保在使用文件对象后正确地关闭文件

使用这个上下文管理器时，我们可以使用with关键字来打开文件并进行读写操作：

with File('file.txt') as file:
    data = file.read()
    print(data)

在这个例子中，Python会首先调用File('file.txt')来创建一个上下文管理器对象，然后调用__enter__方法打开文件，并将返回的文件对象传入代码块中在代码块执行完毕后，Python会自动调用__exit__方法关闭文件

with举例

好的，以下是几个使用 with 关键字的 Python 示例：

使用 with 打开文件并进行读写操作：

with open('file.txt', 'r') as file:
    data = file.read()
    print(data)

使用 with 创建一个临时文件，并在代码块结束时自动删除它：

import tempfile

with tempfile.NamedTemporaryFile() as temp_file:
    print(temp_file.name)  # 输出临时文件的路径名
    # 在这里可以对临时文件进行读写操作

使用 with 锁住一个资源，确保在代码块中独占该资源：

import threading

lock = threading.Lock()

def thread_func():
    with lock:
        # 在这里进行需要互斥访问的代码块操作
        pass

使用 with 创建一个上下文管理器，确保在代码块结束后清理资源：

class Resource:
    def __init__(self):
        self.resource = None
    
    def __enter__(self):
        self.resource = create_resource()  # 创建资源
        return self.resource
    
    def __exit__(self, exc_type, exc_value, traceback):
        cleanup_resource(self.resource)  # 清理资源

with Resource() as resource:
    # 在这里使用 resource 进行操作
    pass

使用 with 创建一个计时器，在代码块结束后输出执行时间：

import time

start_time = time.time()

with Timer() as timer:
    # 在这里执行需要计时的代码块
    pass

end_time = time.time()
print(f"Execution time: {end_time - start_time} seconds")

使用 with 创建一个计数器，在代码块结束后输出计数值：

from collections import defaultdict

def count_words(text):
    counter = defaultdict(int)
    with Counter() as count:
        # 在这里对文本进行分词并计数
        words = text.split()
        for word in words:
            count[word] += 1
    return counter

使用 with 创建一个数据库连接，在代码块结束后自动关闭连接：

import sqlite3

def query_database():
    with sqlite3.connect('database.db') as connection:
        cursor = connection.cursor()
        # 在这里执行数据库查询操作
        pass