파이썬 contextlib 모듈 완전 정복 🧰

파이썬 contextlib 모듈 완전 정복 🧰

contextlib 모듈은 컨텍스트 매니저를 보다 간결하고 효율적으로 작성할 수 있도록 도와주는 유용한 도구입니다. 이 모듈을 활용하면 자원 관리를 보다 명확하게 할 수 있습니다.

1. @contextmanager: 간단한 컨텍스트 매니저 작성

from contextlib import contextmanager

@contextmanager
def open_file(name, mode):
    f = open(name, mode)
    try:
        yield f
    finally:
        f.close()

with open_file('test.txt', 'w') as f:
    f.write('Hello, world!')

2. closing: 종료가 필요한 객체를 컨텍스트 매니저로 사용

from contextlib import closing
from urllib.request import urlopen

with closing(urlopen('http://example.com')) as page:
    for line in page:
        print(line)

3. suppress: 특정 예외 무시하기

from contextlib import suppress

with suppress(FileNotFoundError):
    os.remove('somefile.tmp')

4. redirect_stdout: 출력 리디렉션

from contextlib import redirect_stdout

with open('output.txt', 'w') as f:
    with redirect_stdout(f):
        print('This will go into the file')

요약

• @contextmanager를 사용하여 간단하게 컨텍스트 매니저를 작성할 수 있습니다.
• closing을 통해 종료가 필요한 객체를 with 문과 함께 사용할 수 있습니다.
• suppress를 사용하면 특정 예외를 무시할 수 있습니다.
• redirect_stdout를 활용하여 출력 결과를 파일 등으로 리디렉션할 수 있습니다.