Desde Python 3.3, puede usar la clase ExitStack
del contextlib
módulo para
abrir de forma segura un número arbitrario de archivos .
Puede gestionar un número dinámico de objetos sensibles al contexto, lo que significa que resultará especialmente útil. si no sabe cuántos archivos va a manejar .
De hecho, el caso de uso canónico que se menciona en la documentación es administrar una cantidad dinámica de archivos.
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
Si está interesado en los detalles, aquí hay un ejemplo genérico para explicar cómo 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)
Salida:
0
enter X1
enter X2
enter X3
3
exit X3
exit X2
exit X1
0
[1, 2, 3]