Оскільки 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]