2-4. Context Manager (1)

이번 강의는 앞에 배웠던 필수 문법들이 발판이 되어줄 것이다.
조금은 생소하지만 한번쯤은 들어봐야 할 내용이니 눈 크게 뜨고 한번 보자!

특히 언더바(__ ) 두개로 시작하는 매직메소드는 그동안 눈 마주치면 바로 줄행랑 쳤었는데 오늘은 눈싸움 좀 해봐야겠다~! 👀

파일을 오픈할 때 쓰는 with 문도 써보긴 했지만 이해를 해서 쓰기보다는 복붙해서 쓰기 급급했던 것 같다. 오늘 함 붙자~..

context manager

• 오늘의 키워드
  • Contextlib, __enter__, __exit__, with 기능 직접 구현

context manager?

cm은 원하는 타이밍에 정확하게 리소스를 할당, 제공, 반환하는 역할을 해준다.
가장 대표적인 with 구문을 이해할 수 있어야한다.

try-finally를 이용한 cm

• with 구문이 나오기 전엔 파일을 열고 닫을 때, close를 통해 직접 자원 할당 해제를 해줘야 했다.

file = open('./testfile1.txt','w')
try:
    file.write('context manager test1\ncontextllib test1.')
finally:
    file.close()

with문을 이용한 cm

• with 문은 굳이 close 해주지 않아도 알아서 자원 사용을 종료해준다.

with open('./testfile2.txt','w') as f:
    f.write('context manager test2\ncontextllib test2.')

• 파이썬 내부적으로 with문 안에 close와 관련 내용이 정의되어 있다.
  그 내부적인 구조를 조금만 수정해서 우리만의 with문으로 커스터마이즈 해보자!

cm 정의해보기

• 우선 class를 따로 정의해서 파일을 열고 닫을 때 발생하는 상황을 정의해줄 것이다.

1. __init__ 메소드는 클래스가 정의될 때 자동으로 실행된다.
   파일 이름과 어떤 메소드를 실행할 지를 인자로 받아주었다.

class MyFileWrite():
    def __init__(self,file_name, method):
        print('myFileWrite started : __init__')
        # file.obj = open 함수
        self.file_obj = open(file_name,method)

2. __enter__ 메소드는 with 구문에 진입할 때 자동으로 호출된다.

    def __enter__(self):
            print('MyFileWrite started : __enter__')
            return self.file_obj

3. __exit__ 메소드는 with 구문을 빠져나가기 직전에 실행된다.
   에러가 발생했을 때의 정보들을 인자에 담아서 리턴한다.

   exc_type : 예외 타입
   value: 오류의 내용
   trace_back : 오류를 발생시킨 함수 호출을 역추적한 내용

     def __exit__(self, exc_type, value, trace_back):
            print('MyFileWrite started : __exit__')
            if exc_type:
                # 예외 발생 시 실행됨 - customize 가능함
                print('Logging exception {}'.format(exc_type, value, trace_back))
            self.file_obj.close()

• 전체 코드

class MyFileWrite():
    def __init__(self,file_name, method):
        print('myFileWrite started : __init__')
        # file.obj = open 함수
        self.file_obj = open(file_name,method)

    def __enter__(self):
        print('MyFileWrite started : __enter__')
        return self.file_obj
    
    def __exit__(self, exc_type, value, trace_back):
        print('MyFileWrite started : __exit__')
        if exc_type:
            print('Logging exception {}'.format(exc_type, value, trace_back))
        self.file_obj.close()

with MyFileWrite('./testfile3.txt','w') as f:
    f.write('context manager test3\ncontextllib test3.')

출처: 인프런 - 모두를 위한 파이썬 : 필수 문법 배우기 Feat. 오픈소스 패키지 배포 (Inflearn Original)