Python: contextlib; with

#contextmanager #with #yield #ray

contextlib 의 contextmanager 를 사용하는 목적은

with 문장 안에서 커스텀 Function 을 사용하기 위함이다.

https://docs.python.org/ko/3/library/contextlib.html

contextlib — Utilities for with-statement contexts

보통 많이 사용하는 패턴이

with open('tmp/path/your/desti', 'w') as f:

    f.write("hello python!")

이러면 좋은점이 file 을 열고 닫는 내용을 직접 명시하지 않아도 된다.

ray 를 사용할 때에도 매번 Init() 함수를 호출하고,

끝날 때에는 shutdown() 함수를 꼭 호출해야 한다.

다른 예로는 pyspark 를 사용할 때인데

이때도 init() 과 close() 를 꼭 해줘야 한다.

그래서 이럴때 contextmanager 를 한 번 만들어두고 사용하면 편리할 것 같다는 생각이다.

간단하게 ray 를 예로 들면,

---

import ray

from contextlib import contextmanager

@contextmanager

def ray_cluster(**ray_init_args):

    _ = ray.init(**ray_init_args)

    try:

        yield ray

    finally:

        ray.shutdown()

@ray.remote

def hello_world():

    return "Hello, World!"

with ray_cluster(ignore_reinit_error=True) as ray:

    result = ray.get([hello_world.remote() for i in range(10)])

    print(result)

---

대략 이런 코드 모양이 아닐까 싶다.

@contextmanager 역할이 바로 ray_cluster 함수를 우리 잘 알고있는 open 함수와 비슷하게 만들어 준다.

공식문서를 한 번 천천히 읽어보면서 내 상황에 맞게끔 contextmanager 를 만들어 사용하면 될 듯.