Python에서 context manager를 커스텀하게 만드는 법

파이썬으로 코딩하다 보면, with 문을 엄청 자주 쓰게 되잖아요? 파일 열고 닫을 때, 데이터베이스 연결할 때, 잠깐 써야 하는 자원을 관리할 때 정말 유용하죠. 이게 바로 context manager 덕분인데, 표준 라이브러리에 있는 것만으로는 부족할 때가 있더라고요. 그래서 직접 만들어 써야 할 때가 생기는데, 어떻게 하는 건지 궁금하셨죠? 제가 직접 경험하면서 깨달은 내용을 공유해 드릴게요!

파이썬에서 context manager를 직접 만드는 방법은 크게 두 가지가 있어요. 하나는 클래스를 이용하는 방법이고, 다른 하나는 contextlib.contextmanager라는 데코레이터를 쓰는 방법이죠.

먼저, 클래스를 이용하는 방법부터 알려드릴게요. 이 방법은 좀 더 복잡하지만, 복잡한 로직을 구현할 때 유용해요. 핵심은 __enter__와 __exit__라는 두 개의 특별한 메서드를 클래스 안에 정의하는 거예요. __enter__는 with 문이 시작될 때 실행되고, __exit__는 with 문이 끝나거나 예외가 발생했을 때 실행됩니다. 제가 처음 이걸 봤을 때는 좀 헷갈렸는데, 실제로 코드를 짜보면서 감을 잡았어요.

예를 들어, 데이터베이스 연결을 관리하는 context manager를 만들어 볼게요. __enter__ 메서드에서는 데이터베이스에 연결하고, __exit__ 메서드에서는 연결을 끊는 코드를 작성하면 되죠. 그리고 __exit__ 메서드에는 예외 처리를 위한 세 개의 매개변수가 있는데, 이걸 이용해서 예외가 발생했을 때도 연결을 깔끔하게 끊을 수 있어요. 이 부분은 꼭 기억해 두세요! 예외 처리를 제대로 안 하면, 자원이 제대로 해제되지 않아서 프로그램이 뻗어버리는 경우도 있거든요. 제가 한 번 그랬거든요… ㅠㅠ

다음은 contextlib.contextmanager 데코레이터를 이용하는 방법이에요. 이 방법은 클래스를 이용하는 방법보다 훨씬 간결해요. yield 키워드를 사용해서 __enter__와 __exit__의 역할을 나누는데, 코드가 훨씬 짧고 보기에도 편하죠. 간단한 작업에는 이 방법을 쓰는 게 좋더라고요. 저는 개인적으로 이 방법을 더 자주 사용하는 편이에요. 훨씬 간편하니까요!

자, 그럼 실제 코드를 보여드릴게요. 클래스 기반과 데코레이터 기반, 둘 다 보여드릴 테니 비교해 보세요. 아래 코드에서 1/0 부분을 주석 처리 해제하고 실행해 보시면 예외 처리가 어떻게 동작하는지 확인할 수 있을 거예요.

(여기에 원문의 코드 예제 삽입)

코드를 보시면 아시겠지만, __exit__ 메서드에서 return True를 하면 예외를 잡아서 처리하고, return False를 하면 예외를 다시 발생시켜요. 어떤 방식을 선택할지는 상황에 따라 결정해야겠죠. 그리고 중요한 점! 자원 해제 로직은 항상 finally 블록 안에 넣어야 해요. 그래야 예외가 발생하더라도 자원이 제대로 해제될 수 있답니다. 이 부분 잊지 마세요!

결론적으로, context manager를 직접 구현하는 건 코드를 더 안전하고 효율적으로 만들어주는 중요한 기술이에요. 클래스 기반과 데코레이터 기반, 두 가지 방법을 다 알아두면 상황에 맞게 적절한 방법을 선택해서 사용할 수 있겠죠? 처음에는 좀 어렵게 느껴질 수 있지만, 몇 번 직접 만들어 보면 금방 익숙해질 거예요. 저도 처음에는 막막했지만, 이제는 척척 만들 수 있답니다! 힘내세요! 그리고 궁금한 점 있으면 언제든지 물어보세요!