Несколько переменных в операторе «с»?

Можно ли объявить более одной переменной с помощью 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 и Python 2.7 . Новый withсинтаксис поддерживает несколько контекстных менеджеров:

with A() as a, B() as b, C() as c:
    doSomething(a,b,c)

В отличие от contextlib.nested, это гарантирует , что aи bбудет иметь их __exit__()называеться , даже если C()и это __enter__()метод вызывает исключение.

Вы также можете использовать более ранние переменные в более поздних определениях (ч / т Ахмад ниже):

with A() as a, B(a) as b, C(a, b) as c:
    doSomething(a, c)

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)

SyntaxError: неверный синтаксис

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]

В Python 3.1+ вы можете указать несколько выражений контекста, и они будут обрабатываться так, как если бы несколько withвыражений были вложенными:

with A() as a, B() as b:
    suite

эквивалентно

with A() as a:
    with B() as b:
        suite

Это также означает, что вы можете использовать псевдоним из первого выражения во втором (полезно при работе с соединениями / курсорами БД):

with get_conn() as conn, conn.cursor() as cursor:
    cursor.execute(sql)