关于contextmanager：嵌套Python上下文管理器

Nesting Python context managers

在这个问题中，我定义了一个包含上下文管理器的上下文管理器。完成此嵌套最简单的正确方法是什么？我最终在self.__enter__()中调用self.temporary_file.__enter__()。但是，在self.__exit__中，我很确定如果引发异常，我必须在finally块中调用self.temporary_file.__exit__(type_, value, traceback)。如果self.__exit__出现问题，我应该设置type_，value和traceback参数吗？我检查了contextlib，但找不到任何实用程序来帮助解决这个问题。

来自问题的原始代码：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

import itertools as it
import tempfile

class WriteOnChangeFile:
def __init__(self, filename):
self.filename = filename

def __enter__(self):
self.temporary_file = tempfile.TemporaryFile('r+')
self.f = self.temporary_file.__enter__()
return self.f

def __exit__(self, type_, value, traceback):
try:
try:
with open(self.filename, 'r') as real_f:
self.f.seek(0)
overwrite = any(
l != real_l
for l, real_l in it.zip_longest(self.f, real_f))
except IOError:
overwrite = True
if overwrite:
with open(self.filename, 'w') as real_f:
self.f.seek(0)
for l in self.f:
real_f.write(l)
finally:
self.temporary_file.__exit__(type_, value, traceback)

创建上下文管理器的简单方法是使用contextlib.contextmanager。像这样的东西：

1
2
3
4
5
6

@contextlib.contextmanager
def write_on_change_file(filename):
with tempfile.TemporaryFile('r+') as temporary_file:
yield temporary_file
try:
... some saving logic that you had in __exit__ ...

然后使用with write_on_change_file(...) as f:。
with语句的主体将被执行"而不是"yield。如果要捕获正文中发生的任何异常，请将yield本身包装在try块中。

临时文件将始终正确关闭(当其with块结束时)。