Python에서 컨텍스트 관리자를 사용하는 방법(문 사용)

Python에서 컨텍스트 관리자가 작동하는 방식을 알아보세요. with 문을 익히고, 시스템 리소스를 관리하고, 사용자 정의 컨텍스트 관리자를 작성하세요.

개요

파일 핸들 닫기, 데이터베이스 연결 끊기, 스레드 잠금 해제와 같은 외부 리소스를 적절하게 관리하는 것은 안정적이고 프로덕션에 바로 사용할 수 있는 소프트웨어를 구축하는 데 필수적입니다. 리소스를 해제하지 못하면 메모리 누수, 파일 잠금 오류 및 애플리케이션 중단이 발생할 수 있습니다.

Python의 `with` 문은 컨텍스트 관리자를 생성하여 리소스 정리를 단순화합니다. 컨텍스트 관리자는 코드가 예외를 발생시키거나 일찍 종료되더라도 제어가 블록을 떠날 때 정리 작업이 자동으로 실행되도록 보장합니다. 이렇게 하면 장황하고 반복적인 'try-finally' 블록이 필요하지 않습니다.

컨텍스트 관리자는 `open()`과 같은 내장 작업과 가장 일반적으로 연관되어 있지만 사용자 정의 컨텍스트 관리자를 쉽게 디자인할 수 있습니다. `__enter__` 및 `__exit__` 매직 메서드를 구현하는 클래스를 생성하거나 `contextlib.contextmanager` 데코레이터로 장식된 생성기 함수를 작성하여 이를 빌드할 수 있습니다.

코드 및 실행 출력

이 스크립트는 클래스 구현 및 contextlib 데코레이터를 사용하여 Python에서 사용자 정의 컨텍스트 관리자를 작성하는 방법을 보여줍니다.

context_managers.py

에디터에서 사용해 보세요

from contextlib import contextmanager

# Method 1: Class-based context manager
class CustomResource:
    def __enter__(self):
        print("[Resource] Opening connection...")
        return self

    def do_work(self):
        print("[Resource] Performing action...")

    def __exit__(self, exc_type, exc_val, exc_tb):
        print("[Resource] Closing connection safely.")
        return False

print("--- Class Context Manager ---")
with CustomResource() as resource:
    resource.do_work()

# Method 2: Generator-based context manager (contextlib)
@contextmanager
def simple_timer():
    print("\n[Timer] Started...")
    try:
        yield
    finally:
        print("[Timer] Finished!")

print("\n--- Decorator Context Manager ---")
with simple_timer():
    print("Performing tasks inside context...")

터미널 출력

--- Class Context Manager ---
[Resource] Opening connection...
[Resource] Performing action...
[Resource] Closing connection safely.

--- Decorator Context Manager ---

[Timer] Started...
Performing tasks inside context...
[Timer] Finished!