Wiele zmiennych w instrukcji „z”?

Czy można zadeklarować więcej niż jedną zmienną przy użyciu withinstrukcji w Pythonie?

Coś jak:

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)

... czy problem polega na oczyszczeniu dwóch zasobów jednocześnie?

Jest to możliwe w Pythonie 3 od wersji 3.1 i Pythona 2.7 . Nowa withskładnia obsługuje wiele menedżerów kontekstu:

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

W odróżnieniu od contextlib.nestedgwarantuje to, że ai bbędzie miał swój __exit__()„nazywa nawet jeśli C()czy jest to __enter__()metoda zgłasza wyjątek.

Możesz także użyć wcześniejszych zmiennych w późniejszych definicjach (h / t Ahmad poniżej):

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

contextlib.nested popiera to:

import contextlib

with contextlib.nested(open("out.txt","wt"), open("in.txt")) as (file_out, file_in):

   ...

Aktualizacja:
Aby zacytować dokumentację dotyczącą contextlib.nested:

Przestarzałe od wersji 2.7 : Instrukcja with obsługuje teraz tę funkcję bezpośrednio (bez mylących skłonności do błędów).

Więcej informacji można znaleźć w odpowiedzi Rafała Dowgirda .

Zauważ, że jeśli podzielisz zmienne na linie, musisz użyć ukośników odwrotnych do zawijania nowych linii.

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

Nawiasy nie działają, ponieważ zamiast tego Python tworzy krotkę.

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

Ponieważ w krotkach brakuje __enter__atrybutu, pojawia się błąd (nieopisowy i nie identyfikuje typu klasy):

AttributeError: __enter__

Jeśli spróbujesz użyć as w nawiasach, Python łapie błąd podczas analizy:

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

Błąd składni: nieprawidłowa składnia

Wydaje się, że https://bugs.python.org/issue12782 ma związek z tym problemem.

Myślę, że zamiast tego chcesz to zrobić:

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)

Od wersji Python 3.3 możesz używać klasy ExitStackz contextlibmodułu.

Może zarządzać dynamiczną liczbą obiektów kontekstowych, co oznacza, że ​​okaże się szczególnie przydatny, jeśli nie wiesz, ile plików zamierzasz obsłużyć.

Kanoniczny przypadek użycia wymieniony w dokumentacji zarządza dynamiczną liczbą plików.

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

Oto ogólny przykład:

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)

Wynik:

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]

W Pythonie 3.1+ można określić wiele wyrażeń kontekstowych, które będą przetwarzane tak, jakby withzagnieżdżono wiele instrukcji:

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

jest równa

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

Oznacza to również, że możesz użyć aliasu z pierwszego wyrażenia w drugim (przydatne podczas pracy z połączeniami / kursorami db):

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