您的位置:

contextlib模块详解

一、contextlib介绍

Python的contextlib模块是Python内置库中的一个模块,可以在编写Python代码时提供非常实用的功能。该模块提供了一个用于创建及管理RAII-style上下文(context)以及一些可以方便任务执行的辅助函数。

RAII-style是Resource Acquisition Is Initialization的缩写,用于C++及其他编程语言的资源管理技术,用于确保程序在操作完资源后,能够及时及时地释放它们。


import contextlib

@contextlib.contextmanager
def my_context():
    # setup
    yield
    # teardown

上面的代码中,我们首先导入了contextlib模块,然后定义了一个简单的上下文管理器my_context,谨记的是,在编写上下文管理器时你需要保证强制使用try/finally语句块实现contextmanager装饰器的效果。

二、with语句

with语句是Python在2.5版本之后引入的一个语法结构。它可以为用户定义的代码块创建一个运行的上下文环境,并且当代码块执行完毕以后,能够自动清理资源。

下面是一个简单的使用with语句创建上下文环境的例子:


try:
    with open("file.txt") as f:
        data = f.read()
except:
    print("failed to read the file..")

上面的例子中,我们打开了一个文件,并使用with语句创建了一个上下文环境。当代码块执行完毕后,with语句会自动关闭该文件,无需手动调用close()方法。

三、closing函数

closing()是在Python内置的contextlib模块中提供的一个功能函数。该函数可以把支持上下文管理协议且没有实现__exit__()方法的对象包装成一个上下文管理器对象。


import contextlib
from urllib.request import urlopen

with contextlib.closing(urlopen("http://www.baidu.com")) as page:
    for line in page:
        print(line)

上面的代码中,我们使用closing函数封装了urlopen返回的HTTPResponse对象,这种用法比较有用,因为HTTPResponse对象不会自动释放它的网络连接。

四、redirect_stdout函数

使用redirect_stdout,我们可以重定向输出到一个文件,stdout或别的流对象,然后再变回直接输出到终端。


import contextlib
import io

with open("file.txt", "w") as f:
    with contextlib.redirect_stdout(f):
        print("Hello, world!")

上面的代码中,我们将print输出到了文件“file.txt”中,而没有输出到终端。这种用法用于记录特定的输出日志信息非常有用。

五、Example: Timer类

为了更好地理解contextlib的特性,我们可以看看它提供了很多什么功能。下面,我将展示一个使用contextlib实现的Timer类,以便更好地理解其中的细节:


import contextlib
import time

class Timer:
    def __init__(self, name):
        self.name = name
        self.start_time = None

    def __enter__(self):
        self.start_time = time.perf_counter()
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        elapsed_time = time.perf_counter() - self.start_time
        print(f"{self.name} took {elapsed_time:.6f} seconds")

with Timer("test"):
    time.sleep(1)

上面的代码中,我们使用Timer类实现了一个简单计时器。首先,我们使用__enter__函数开始计时,并将self传递回上下文管理器中。当用户代码块完成后,会自动调用__exit__函数停止计时,并打印出花费的时间。这样我们就创建了一个安全、可重用的计时器。