Как удалить диспетчер контекста из ExitStack

Question

У меня есть длительный процесс, называемый Updater, который имеет обновления (к системе ETL), представленные на него. У обновлений есть требования к ресурсам, которым управляет добавление диспетчера контекстов к ExitStack Updater. Некоторые обновления будут включать новую конфигурацию, что означает, что выделенные ресурсы должны быть освобождены из стека, и будет добавлена ​​новая конфигурация ресурса. Мне нужно что-то вроде:

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 

Ниже приведен пример того, что я стал работать, но она опирается на доступ к защищенным членам. Я надеялся, что там может быть меньше «грязное» решением, чем это:

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) 

Edit: Добавлен известным решением

Answer 1

contextlib.ExitStack поддерживает только выход всех менеджеров контекста сразу. Вы не можете создавать контекстные менеджеры по отдельности. Если вы хотите это сделать, вы должны следить за своими менеджерами контекста за чем-то другим, кроме ExitStack.

Answer 2

Вы должны расширить ExitStack с вашим собственным pop -метода:

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)