Python Academy · 강의

enter 및 exit 프로토콜

든더 프로토콜을 사용해 컨텍스트 관리자를 만듭니다.

__enter__ 및 __exit__ 프로토콜은(는) CoddyKit의 무료 Python Academy 강의입니다. 이것은 4개 중 2번째 강의입니다. 아래에서 전체 강의를 무료로 읽을 수 있으며, 내장 코드 에디터와 24/7 AI 튜터와 함께 브라우저에서 직접 실습할 수 있습니다. 이 강의는 Python Academy 학습 경로의 일부이며, 진행 상황이 웹과 CoddyKit 앱에 동기화됩니다. Python Academy 강의에는 총 4개의 강의가 포함되어 있습니다.

컨텍스트 관리자 프로토콜

__enter__와 __exit__를 정의하는 모든 객체는 컨텍스트 관리자이며 with와 함께 사용할 수 있습니다.

class Managed:
    def __enter__(self):
        print("entering")
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        print("exiting")
        return False

with Managed() as m:
    print("inside")

enter 반환 값

__enter__는 with 블록이 시작될 때 호출됩니다. 반환 값은 as 뒤의 변수에 할당됩니다.

class Resource:
    def __enter__(self):
        self.value = 42
        return self  # bound to 'r'

    def __exit__(self, *args):
        self.value = None
        return False

with Resource() as r:
    print(r.value)  # 42

exit 인수

__exit__(exc_type, exc_val, exc_tb)는 오류가 발생한 경우 예외 정보를 받습니다. 예외가 없으면 세 값 모두 None입니다.

class Safe:
    def __enter__(self):
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        if exc_type is ValueError:
            print(f"Suppressed ValueError: {exc_val}")
            return True   # suppress the exception
        return False      # re-raise anything else

with Safe():
    raise ValueError("oops")  # suppressed

예외 억제하기

예외를 억제하려면 __exit__에서 True를 반환하십시오. 예외가 전파되도록 하려면 False를 반환하거나 아무것도 반환하지 마십시오.

class Ignore:
    def __enter__(self): return self
    def __exit__(self, *args): return True  # swallow all exceptions

with Ignore():
    1 / 0

print("continues here")

데이터베이스 연결 예시

대표적인 사용 사례는 진입할 때 연결을 확보하고, 종료할 때 커밋 또는 롤백한 뒤 닫는 것입니다.

class DBConn:
    def __enter__(self):
        self.conn = connect_db()
        return self.conn

    def __exit__(self, exc_type, exc_val, exc_tb):
        if exc_type:
            self.conn.rollback()
        else:
            self.conn.commit()
        self.conn.close()
        return False

타이머 컨텍스트 관리자

__enter__에서 시작 시각을 기록하고 __exit__에서 차이를 계산하여 경과 시간을 측정합니다.

import time

class Timer:
    def __enter__(self):
        self.start = time.perf_counter()
        return self

    def __exit__(self, *args):
        self.elapsed = time.perf_counter() - self.start
        print(f"Elapsed: {self.elapsed:.3f}s")
        return False

with Timer():
    sum(range(1_000_000))

컨텍스트 관리자 중첩하기

여러 컨텍스트 관리자를 하나의 with 문에 결합할 수 있습니다.

with open("in.txt") as src, open("out.txt", "w") as dst:
    dst.write(src.read())

재사용 가능한 관리자와 일회용 관리자

__enter__가 호출될 때마다 새로운 리소스를 할당하는 컨텍스트 관리자는 재사용할 수 있습니다. 내부 상태를 소진하는 관리자는 재사용할 수 없을 수 있습니다. 예를 들어 한 번만 획득할 수 있는 잠금이 있습니다.

class Reusable:
    def __enter__(self):
        return []
    def __exit__(self, *a):
        return False

for _ in range(3):
    with Reusable() as lst:
        lst.append(1)
        print(lst)  # [1] each time

aenter와 aexit (비동기)

비동기 컨텍스트 관리자의 경우 __aenter__와 __aexit__ 코루틴을 정의하고 async with를 사용하십시오.

class AsyncDB:
    async def __aenter__(self):
        self.conn = await connect()
        return self.conn

    async def __aexit__(self, *args):
        await self.conn.close()

# async with AsyncDB() as conn:
#     ...

파일 객체 래핑하기

파이썬 파일 객체는 이미 컨텍스트 관리자 프로토콜을 구현하므로 f.close()를 직접 호출하지 마십시오.

with open("data.txt", "r") as f:
    data = f.read()
# file is closed here automatically
print(f.closed)  # True

잠금 구현하기

스레드 잠금은 컨텍스트 관리자입니다. __enter__는 잠금을 획득하고 __exit__는 잠금을 해제합니다.

import threading

lock = threading.Lock()

def update_shared():
    with lock:
        # critical section
        pass

빠른 확인

예외를 억제하려면 __exit__는 무엇을 반환해야 하나요?

복습

컨텍스트 관리자를 만들려면 __enter__와 __exit__를 구현하십시오. __enter__에서 리소스를 반환하고, __exit__에서 정리와 예외 억제를 처리하십시오.

자주 묻는 질문