Python contextlib 怎样把多个资源组合关闭

当程序要同时打开多个文件、锁或临时资源，而且资源数量由配置或输入决定时，逐层嵌套 with 很快会变得难维护。Python 的 contextlib.ExitStack 可以把这些资源登记到同一个退出栈中：后登记的资源先关闭，后续初始化失败时，前面已经登记的资源仍会自动收尾。

一、资源数量可变时用 ExitStack 统一登记

ExitStack 的核心不是“把所有对象放进列表”，而是保存每个资源对应的退出动作。下面的例子根据文件名列表动态打开文件，任意一次 open() 失败，已经登记的文件仍会由同一个 with 释放。

from contextlib import ExitStack

def read_many(paths):
    # 文件数量由输入决定，不能把 with 层数写死。
    with ExitStack() as stack:
        files = [stack.enter_context(open(path, encoding="utf-8"))
                 for path in paths]
        # 这里使用资源；后续 open 失败时，已打开文件仍会按逆序关闭。
        return [file.read() for file in files]

enter_context(cm) 会调用上下文管理器的进入逻辑，并把退出逻辑压入栈中。它返回的是 __enter__() 的结果，所以列表里的元素是可读文件对象，而不是原始上下文管理器。

二、enter_context 和 callback 都按登记逆序收尾

如果资源本身支持 with，优先使用 enter_context()。只有释放动作是一个普通函数时，才用 callback(function, *args) 登记。两者混用时仍遵守同一条规则：最后登记的动作最先执行，这等价于多层嵌套的 with。

from contextlib import ExitStack

def release_cache(cache):
    # 普通对象没有 __exit__ 时，用显式函数描述释放动作。
    cache.clear()

with ExitStack() as stack:
    first = stack.enter_context(open("first.txt", encoding="utf-8"))
    cache = {}
    stack.callback(release_cache, cache)
    second = stack.enter_context(open("second.txt", encoding="utf-8"))
    # 退出时依次关闭 second、执行 release_cache、关闭 first。
    print(first.readline(), second.readline())

不要把“注册 callback”误认为“立刻执行 cleanup”。回调只在 ExitStack 退出或显式调用 close() 时运行；对象被垃圾回收并不会替你关闭这个栈。

三、部分初始化失败时，让栈接住异常路径

批量打开资源时，真正容易漏的是中途失败：第一个文件已经打开，第二个文件因为权限或路径错误抛异常。如果每个对象都单独维护 try/finally，清理责任会散落在多处；将成功获取的资源立即登记，退出栈就能覆盖这条路径。

from contextlib import ExitStack

def open_required(paths):
    with ExitStack() as stack:
        handles = []
        for path in paths:
            # 只有 open 成功后才加入 handles，退出动作由 stack 管理。
            handle = stack.enter_context(open(path, "rb"))
            handles.append(handle)
        # 只有全部资源准备好才返回；异常会先关闭已登记资源。
        return handles

这个函数不应直接把 handles 返回给调用者：with 结束时它们已经关闭。若资源需要交给外层继续使用，应先把栈的清理责任一起转移。

四、成功交接时用 pop_all 转移清理责任

pop_all() 会把当前栈中的退出回调移到一个新的栈，并清空原栈。典型场景是“先完成一组资源初始化，确认成功后再由调用方负责关闭”。如果初始化失败，不调用 pop_all()，原 with 仍会自动清理。

from contextlib import ExitStack

def acquire(paths):
    with ExitStack() as stack:
        files = [stack.enter_context(open(path, encoding="utf-8"))
                 for path in paths]
        # 所有资源准备完成后，转移关闭责任给返回值的持有者。
        owner_stack = stack.pop_all()
        return files, owner_stack

files, owner_stack = acquire(["a.txt", "b.txt"])
try:
    print(files[0].read())
finally:
    # 外层明确拥有清理责任，close 仍按登记逆序执行。
    owner_stack.close()

pop_all() 不是“取消关闭”，而是“换一个 owner 关闭”。如果返回文件却丢掉新的 stack，就等于丢失清理责任。

五、组合关闭的检查清单

可以把实现压缩成一句判断：资源是否已成功获得，就立即登记；资源是否仍由当前函数负责，就保留当前栈；资源是否已经交给外层，就用 pop_all() 明确转移。这样既能保证异常路径释放，也不会在成功交接后重复关闭。

相关问题

ExitStack 会按打开顺序关闭文件吗？

不会。它按登记逆序执行退出动作，最后登记的文件先关闭，这正是嵌套 with 的自然行为。

没有 close 方法的资源能放进 ExitStack 吗？

可以，只要能提供一个明确的释放函数，就用 stack.callback() 登记；若没有可靠的释放动作，不应把它伪装成已受管理资源。