python - 如何从 ExitStack 中删除上下文管理器

Question

我有一个名为“更新程序”的长期运行进程，它已提交更新(到 ETL 系统)。更新具有通过将上下文管理器添加到更新程序的 ExitStack 来管理的资源要求。某些更新将包括新配置，这意味着必须从堆栈中释放受影响的资源，并且将添加该资源的新配置版本。我需要类似的东西:

with ExitStack() as stack:
    ctx_manager = open("file.txt")
    f = stack.enter_context(ctx_manager)
    ...
    ctx_pop(ctx_manager, stack)  # remove the given context manager from the stack

下面是我已经完成的工作的示例，但它依赖于访问 protected 成员。我希望可能有一个比这更“肮脏”的解决方案:

def ctx_pop(cm, stack):
    for item in stack._exit_callbacks:
        if item.__self__ is cm:
            break
    else:
        raise KeyError(repr(cm))
    stack._exit_callbacks.remove(item)
    item(None, None, None)

编辑:添加已知解决方案

Answer 1

您必须使用自己的 pop 方法扩展 ExitStack:

from contextlib import ExitStack
from collections import deque

class ExitStackWithPop(ExitStack):
    def pop(self, cm):
        callbacks = self._exit_callbacks
        self._exit_callbacks = deque()
        found = None
        while callbacks:
            cb = callbacks.popleft()
            if cb.__self__ == cm:
                found = cb
            else:
                self._exit_callbacks.append(cb)
        if not found:
            raise KeyError("context manager not found")
        found(None, None, None)