Несколько переменных в операторе with?
-
23-08-2019 - |
Вопрос
Можно ли объявить более одной переменной, используя with
заявление на Python?
Что-то вроде:
from __future__ import with_statement
with open("out.txt","wt"), open("in.txt") as file_out, file_in:
for line in file_in:
file_out.write(line)
...или проблема в очистке двух ресурсов одновременно?
Решение
Это возможно в Python 3 начиная с версии 3.1 и Питон 2.7.Новый with
синтаксис поддерживает несколько менеджеров контекста:
with A() as a, B() as b, C() as c:
doSomething(a,b,c)
в отличие от contextlib.nested
, это гарантирует, что a
и b
будут иметь свои __exit__()
вызывается, даже если C()
или это __enter__()
метод вызывает исключение.
Другие советы
contextlib.nested
поддерживает это:
import contextlib
with contextlib.nested(open("out.txt","wt"), open("in.txt")) as (file_out, file_in):
...
Обновлять:
Процитируем документацию, касающуюся contextlib.nested
:
Устарело с версии 2.7.:Заявление with теперь поддерживает это функционал напрямую (без путаницы склонных к ошибкам причуд).
Видеть Ответ Рафала Даугирда Чтобы получить больше информации.
Обратите внимание: если вы разбиваете переменные на строки, для переноса новых строк необходимо использовать обратную косую черту.
with A() as a, \
B() as b, \
C() as c:
doSomething(a,b,c)
Круглые скобки не работают, поскольку вместо этого Python создает кортеж.
with (A(),
B(),
C()):
doSomething(a,b,c)
Поскольку в кортежах отсутствует __enter__
атрибут, вы получите сообщение об ошибке (неописательное и не идентифицирующее тип класса):
AttributeError: __enter__
Если вы попытаетесь использовать as
внутри круглых скобок Python обнаруживает ошибку во время анализа:
with (A() as a,
B() as b,
C() as c):
doSomething(a,b,c)
Ошибка синтаксиса:неверный синтаксис
https://bugs.python.org/issue12782 кажется, связано с этой проблемой.
Я думаю, вы хотите сделать это вместо этого:
from __future__ import with_statement
with open("out.txt","wt") as file_out:
with open("in.txt") as file_in:
for line in file_in:
file_out.write(line)
Начиная с 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
Вот общий пример:
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(stack._exit_callbacks)
nums = [stack.enter_context(x) for x in xs]
print(stack._exit_callbacks)
print(stack._exit_callbacks)
print(nums)
Выход:
deque([])
enter X1
enter X2
enter X3
deque([<function ExitStack._push_cm_exit.<locals>._exit_wrapper at 0x7f5c95f86158>, <function ExitStack._push_cm_exit.<locals>._exit_wrapper at 0x7f5c95f861e0>, <function ExitStack._push_cm_exit.<locals>._exit_wrapper at 0x7f5c95f86268>])
exit X3
exit X2
exit X1
deque([])
[1, 2, 3]