[Python] functools 의 partial | LIM

🧐 functools 모듈이란?

functools 모듈을 몇 가지 고위 함수를 통합한다. 그 중 가장 널리 알려진 함수가 reduce() 함수이다. functools에서 제공하는 나머지 함수 중 partial() 및 이의 변형인 partialmethod() 함수가 매우 유용하다고 한다. 

 

✔️ run_in_executor 에 이용한 functools.partial

functools.partial()은 함수를 부분적으로 실행할 수 있게 해주는 고위함수이다. 어떤 함수가 있을 때 partial()을 적용하면 원래 함수의 일부 인수를 고정한 콜러블을 생성한다. 이 기법은 하나 이상의 인수를 받는 함수를 그보다 적은 인수를 받는 콜백 함수를 사용하는 API에 사용하고자 할 때 유용하다. 

 

예를 들어 비동기 처리할 때 네이티브 코루틴 안에서 블로킹 I/O 함수(requests.post)를 실행하기 위해 이벤트 루프의 run_in_executor를 사용했는데 run_in_executor의 함수는 이렇게 생겼다.

 

run_in_executor는 비동기를 고려하지 않고 만들어진 라이브러리를 비동기로 이용할 수 있게 해준다.

 

def run_in_executor(self, executor, func, *args):
    self._check_closed()
    if self._debug:
        self._check_callback(func, 'run_in_executor')
    if executor is None:
        executor = self._default_executor
        # Only check when the default executor is being used
        self._check_default_executor()
        if executor is None:
            executor = concurrent.futures.ThreadPoolExecutor(
                thread_name_prefix='asyncio'
            )
            self._default_executor = executor
    return futures.wrap_future(
        executor.submit(func, *args), loop=self)

저 func 인자 안에 requests.post(url, params=params) 를 실행하는 함수를 넣고 싶었다. 이때 functools.partial()을 사용했다.

 

partial 함수는 기존 함수와 동일하지만 파라미터를 미리 정해준 또 다른 함수라고 생각하면 된다.

 

따라서 나는 다음과 같이 사용했다.

async def post_test(type, day):
    loop = asyncio.get_event_loop()
    await loop.run_in_executor(None, 
    	functools.partial(requests.post, url, params={'type': type, 'day': day}))

 

즉 functools.partial을 이용해서

requests.post(url, params={'settlementType': settlementType, 'tradingDay': tradingDay})

위와 같은 기능을 실행하게끔 만든 것이다.