ExitStack에서 컨텍스트 관리자를 제거하는 방법

Question

Updater라는 장기 실행 프로세스가 있으며이 프로세스에는 업데이트 (ETL 시스템에 대한 업데이트)가 제출되었습니다. 업데이트에는 컨텍스트 관리자를 Updater의 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 

다음은 내가 작업 한 예이지만 보호 된 멤버에 액세스하는 방법에 의존합니다.

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

contextlib.ExitStack은 한 번에 모든 컨텍스트 관리자를 종료 지원 나는 이것보다 덜 '더러운'해결책이 될 수있을 기대했다. 컨텍스트 관리자는 개별적으로 팝 할 수 없습니다. 이 작업을 수행하려면 컨텍스트 관리자를 ExitStack이 아닌 다른 항목으로 추적해야합니다.

Answer 2

당신은 당신의 자신의 pop -method와 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)