ExitStack — 동적 컨텍스트 매니저 스택

지난 글에서 suppress, redirect_stdout 같은 contextlib 유틸리티를 살펴봤다. 이번에는 contextlib의 가장 강력한 도구인 ExitStack을 다룬다. 컨텍스트 매니저의 개수가 컴파일 시점에 정해지지 않는 경우를 우아하게 처리한다.

ExitStack이 필요한 이유

with open(a) as f1, open(b) as f2: 처럼 중첩 with 문은 파일 수가 고정일 때는 잘 동작한다. 그런데 파일 이름 목록이 런타임에 결정된다면?

# 이렇게는 못 쓴다 — 개수가 동적이라서
files = [open(fn) for fn in filenames]  # 예외 발생 시 앞 파일이 닫히지 않음

수동으로 try/finally를 중첩하면 코드가 복잡해진다. ExitStack은 이 문제를 깔끔하게 해결한다.

기본 사용법

from contextlib import ExitStack

filenames = ["a.csv", "b.csv", "c.csv"]

with ExitStack() as stack:
    files = [stack.enter_context(open(fn)) for fn in filenames]
    # 블록 종료 시 등록 역순(LIFO)으로 __exit__ 호출
    process(files)

enter_context(cm)은 cm.__enter__()를 호출하고 반환값을 돌려주면서, cm을 스택에 등록한다. with 블록을 나가면 등록된 매니저들이 LIFO 순서로 __exit__을 호출한다.

callback — 임의 정리 함수 등록

컨텍스트 매니저가 아닌 일반 함수도 스택에 등록할 수 있다.

with ExitStack() as stack:
    conn = db.connect()
    stack.callback(conn.close)        # 함수 참조만 전달
    stack.callback(log.info, "done")  # 인수도 전달 가능

    do_work(conn)
# 블록 종료 시: log.info("done") → conn.close() 역순 실행

pop_all — 소유권 이전

pop_all()은 현재 스택의 등록 목록을 새 ExitStack으로 이전하고 기존 스택을 비운다. 팩토리 함수에서 자원을 안전하게 반환할 때 유용하다.

def open_multiple(filenames):
    with ExitStack() as cleanup:
        files = [cleanup.enter_context(open(fn)) for fn in filenames]
        # 여기까지 오면 모든 파일이 열린 것
        # cleanup 스택을 caller 에게 이전 (소유권 넘김)
        return files, cleanup.pop_all()

files, stack = open_multiple(["a.csv", "b.csv"])
with stack:
    process(files)
# 여기서 파일들 자동 닫힘

pop_all()이 없으면 open_multiple에서 예외가 나면 cleanup이 with 블록 종료로 파일을 닫아주지만, 정상 반환 시엔 정리 책임이 호출자에게 넘어가야 한다.

예외 처리

ExitStack은 각 __exit__ 호출의 예외를 개별 처리한다. 하나가 실패해도 나머지 __exit__는 계속 실행된다.

from contextlib import ExitStack, suppress

with ExitStack() as stack:
    # 특정 매니저의 예외만 억제
    stack.enter_context(suppress(OSError))
    stack.enter_context(open("data.txt"))
    stack.enter_context(open("log.txt", "w"))
    # OSError 는 억제, 나머지 예외는 전파

AsyncExitStack

비동기 환경에서는 contextlib.AsyncExitStack을 쓴다. aenter_context와 async with를 사용한다는 점만 다르다.

from contextlib import AsyncExitStack

async def main():
    async with AsyncExitStack() as stack:
        conn = await stack.enter_async_context(async_db_connect())
        await do_async_work(conn)

요약

• ExitStack: 런타임에 결정된 개수의 컨텍스트 매니저를 LIFO로 관리
• enter_context(cm): 스택에 등록 + __enter__() 반환값 반환
• callback(fn, *args): 일반 함수를 정리 콜백으로 등록
• pop_all(): 스택 소유권을 새 ExitStack으로 이전
• AsyncExitStack: 비동기 버전

---

지난 글: contextlib.suppress와 redirect

다음 글: 비동기 컨텍스트 매니저

읽어주셔서 감사합니다. 😊