Кілька змінних у операторі "з"?

Чи можливо оголосити більше однієї змінної за допомогою a 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 з v3.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__()метод викликає виняток.

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

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 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

Це також означає, що ви можете використовувати псевдоніми з першого виразу в другому (корисно при роботі з db-з'єднаннями / курсорами):

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