So entfernen Sie einen Kontextmanager aus einem ExitStack

Question

Ich habe einen lang laufenden Prozess namens Updater, der über Updates (an ein ETL-System) verfügt. Die Aktualisierungen verfügen über Ressourcenanforderungen, die durch Hinzufügen eines Kontextmanagers zum ExitStack des Updaters verwaltet werden. Einige Updates beinhalten eine neue Konfiguration, was bedeutet, dass betroffene Ressourcen vom Stack freigegeben werden müssen und eine neu konfigurierte Version der Ressource hinzugefügt wird. Ich brauche so etwas wie:

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 

Im Folgenden ist ein Beispiel für etwas, das ich zu arbeiten bekommen haben, aber es stützt sich auf geschützte Mitglieder zugreifen. Ich hatte gehofft, es könnte eine weniger ‚schmutzige‘ Lösung als diese:

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: Hinzugefügt bekannte Lösung

Answer 1

contextlib.ExitStack nur auf einmal Verlassen alle Kontext-Manager unterstützt. Sie können Kontextmanager nicht einzeln aufrufen. Wenn Sie dies tun möchten, sollten Sie Ihre Kontextmanager mit etwas anderem als einem ExitStack verfolgen.

Answer 2

Sie haben ExitStack mit Ihrem eigenen pop -Methode erweitern:

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)