'with'语句中的多个变量？

Question

是否可以在 Python 中使用with 语句声明多个变量？

类似：

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)

...还是同时清理两个资源的问题？

Answer 1

在Python 3 since v3.1 和Python 2.7 中是可能的。新的with syntax 支持多个上下文管理器：

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

与contextlib.nested 不同，这保证a 和b 将调用它们的__exit__()，即使C() 或其__enter__() 方法引发异常。

您还可以在以后的定义中使用较早的变量（h/t Ahmad 下面）：

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

从 Python 3.10 开始，you can use parentheses：

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

Answer 2

请注意，如果将变量拆分为行，在 Python 3.10 之前，您必须使用反斜杠来换行。

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: invalid syntax

什么时候能解决这个问题？

在https://bugs.python.org/issue12782 中跟踪此问题。

Python 在PEP 617 中宣布他们将用新的解析器替换原来的解析器。因为 Python 的原始解析器是 LL(1)，它在“多个上下文管理器”with (A(), B()): 和“值元组”with (A(), B())[0]: 之间cannot distinguish。

新的解析器可以正确解析多个用括号括起来的上下文管理器。新解析器已在 3.9 中启用。据报道，在 Python 3.10 中删除旧的解析器之前，这种语法仍然会被拒绝，并且在3.10 release notes 中报告了这种语法变化。但在我的测试中，它也适用于 trinket.io 的 Python 3.9.6。

Answer 3

contextlib.nested 支持这个：

import contextlib

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

   ...

更新：
引用文档，关于contextlib.nested：

自 2.7 版起已弃用：with 语句现在支持此功能 直接的功能（没有令人困惑的容易出错的怪癖）。

请参阅Rafał Dowgird's answer 了解更多信息。

Answer 4

从 Python 3.3 开始，您可以使用来自 contextlib 模块的类 ExitStack。

它可以管理 动态 数量的上下文感知对象，这意味着如果您不知道要处理多少个文件，它会特别有用。

文档中提到的规范用例是管理动态数量的文件。

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]

Answer 5

我想你想这样做：

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)

Answer 6

从 Python 3.10 开始，Parenthesized context managers 有了一个新功能，它允许以下语法：

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

Answer 7

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

Answer 8

您还可以separate 创建上下文管理器（__init__ 方法）并输入上下文（__enter__ 方法）以提高可读性。所以不要写这段代码：

with Company(name, id) as company, Person(name, age, gender) as person, Vehicle(brand) as vehicle:
    pass

你可以写这段代码：

company = Company(name, id)
person = Person(name, age, gender)
vehicle = Vehicle(brand)

with company, person, vehicle:
    pass

请注意，在with 语句之外创建上下文管理器给人的印象是，创建的对象也可以在语句之外进一步使用。如果您的上下文管理器不是这样，则错误印象可能与可读性尝试相对应。

documentation 说：

大多数上下文管理器的编写方式意味着它们只能在 with 语句中有效使用一次。这些一次性上下文管理器必须在每次使用时重新创建 - 再次尝试使用它们会触发异常或无法正常工作。

这个常见的限制意味着通常建议直接在使用它们的 with 语句的标题中创建上下文管理器。