如何用Python编写实现上下文管理器的代码？

本文共计691个文字，预计阅读时间需要3分钟。

问题：你想自己实现一个新的上下文管理器，以便使用with语句。解决方案：实现一个新上下文管理器的最简单方法就是使用`contexlib`模块中的`@contextmanager`装饰器。下面是一个实现示例：

问题

你想自己去实现一个新的上下文管理器，以便使用with语句。

解决方案

实现一个新的上下文管理器的最简单的方法就是使用 contexlib 模块中的 @contextmanager 装饰器。 下面是一个实现了代码块计时功能的上下文管理器例子：

import time from contextlib import contextmanager @contextmanager def timethis(label): start = time.time() try: yield finally: end = time.time() print('{}: {}'.format(label, end - start)) # Example use with timethis('counting'): n = 10000000 while n > 0: n -= 1

在函数 timethis() 中，yield 之前的代码会在上下文管理器中作为 __enter__() 方法执行， 所有在 yield 之后的代码会作为 __exit__() 方法执行。 如果出现了异常，异常会在yield语句那里抛出。