<Python> Decorator

Decorator는 이미 만든 함수를 수정하지 않고, 함수 주변을 감싸는 방식으로 함수에 추가 기능을 구현한다.

def trace(func):
    # wrapper는 아래처럼 가변 인수로 만들 수 있다.
    # 가변 인수가 아닌 경우, 원래 함수의 parameter 형태와 일치해야 한다.
    def wrapper(*args, **kwargs):
        ret = func(*args, **kwargs)
        print('{0}(args={1}, kwargs={2}) -> {3}'.format(func.__name__, args, kwargs, ret))
        # 원래 함수가 return이 필요한 경우에는 wrapper도 return이 필요하다.
        return ret
    return wrapper

@trace
def get_max(*args):
    return max(args)

print(get_max(30, 40, 50))

 

매개변수가 있는 decorator는 아래와 같이 만든다. (함수를 한번 더 포장한다)

import functools

# 매개 변수가 있는 Decorator는 wrapper 위에 @functools.wraps(func)를 추가한다.
def hello(t):
    def trace(func):
        # wrapper는 아래처럼 가변 인수로 만들 수 있다.
        # 가변 인수가 아닌 경우, 원래 함수의 parameter 형태와 일치해야 한다.
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            print('hello'*t)
            ret = func(*args, **kwargs)
            print('{0}(args={1}, kwargs={2}) -> {3}'.format(func.__name__, args, kwargs, ret))
            # 원래 함수가 return이 필요한 경우에는 wrapper도 return이 필요하다.
            return ret
        return wrapper
    return trace

@hello(3)
def get_max(*args):
    return max(args)

print(get_max(30, 40, 50))