Python 3.3以降ExitStack
、contextlib
モジュールのクラスを使用して
、任意の数のファイルを安全に開くことができます。
動的な数のコンテキスト対応オブジェクトを管理できます。つまり、処理するファイルの数がわからない場合に特に役立ちます。
実際、ドキュメントに記載されている正規のユースケースでは、動的な数のファイルを管理しています。
with ExitStack() as stack:
files = [stack.enter_context(open(fname)) for fname in filenames]
# All opened files will automatically be closed at the end of
# the with statement, even if attempts to open files later
# in the list raise an exception
詳細に興味がある場合は、ExitStack
動作方法を説明するための一般的な例を以下に示します。
from contextlib import ExitStack
class X:
num = 1
def __init__(self):
self.num = X.num
X.num += 1
def __repr__(self):
cls = type(self)
return '{cls.__name__}{self.num}'.format(cls=cls, self=self)
def __enter__(self):
print('enter {!r}'.format(self))
return self.num
def __exit__(self, exc_type, exc_value, traceback):
print('exit {!r}'.format(self))
return True
xs = [X() for _ in range(3)]
with ExitStack() as stack:
print(len(stack._exit_callbacks)) # number of callbacks called on exit
nums = [stack.enter_context(x) for x in xs]
print(len(stack._exit_callbacks))
print(len(stack._exit_callbacks))
print(nums)
出力:
0
enter X1
enter X2
enter X3
3
exit X3
exit X2
exit X1
0
[1, 2, 3]