컨텍스트 매니저 에러 처리

Python code

from contextlib import contextmanager, suppress

# 예외를 변환하는 컨텍스트 매니저
@contextmanager
def error_handler(error_map: dict):
    try:
        yield
    except tuple(error_map.keys()) as e:
        new_error_cls = error_map[type(e)]
        raise new_error_cls(str(e)) from e

# 사용
class APIError(Exception): pass

try:
    with error_handler({ConnectionError: APIError, TimeoutError: APIError}):
        raise ConnectionError("서버 다운")
except APIError as e:
    print(f"API 에러로 변환됨: {e}")

# suppress: 특정 예외 무시
import os
with suppress(FileNotFoundError):
    os.remove("없는파일.txt")
    # FileNotFoundError가 발생해도 무시됨
print("파일 삭제 시도 완료")

# 리소스 정리 보장
@contextmanager
def managed_resource(name):
    print(f"  {name} 획득")
    try:
        yield name
    except Exception as e:
        print(f"  {name} 에러 처리: {e}")
        raise  # 예외 재발생
    finally:
        print(f"  {name} 해제")

try:
    with managed_resource("DB") as r:
        raise RuntimeError("쿼리 실패")
except RuntimeError:
    print("외부에서 처리")