Desde o Python 3.3, você pode usar a classe ExitStack
do contextlib
módulo para
abrir com segurança um número arbitrário de arquivos .
Ele pode gerenciar um número dinâmico de objetos sensíveis ao contexto, o que significa que será especialmente útil se você não souber quantos arquivos manipulará .
De fato, o caso de uso canônico mencionado na documentação está gerenciando um número dinâmico de arquivos.
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
Se você estiver interessado nos detalhes, aqui está um exemplo genérico para explicar como ExitStack
funciona:
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)
Resultado:
0
enter X1
enter X2
enter X3
3
exit X3
exit X2
exit X1
0
[1, 2, 3]