Decorator

함수를 받아 명령을 추가한 뒤 이를 다시 함수의 형태로 반환하는 함수

• 함수의 내부를 수정하지 않고 기능에 변화를 주고 싶을 때 사용
• 일반적으로 함수의 전처리나 후처리에 대한 필요가 있을 때 사용
• 데코레이터를 이용하면 반복을 줄이고 메소드나 함수의 책임을 확장 가능

Decorator 구조

• 함수로 만들 때

def out_func(func):  # 기능을 추가할 함수를 인자로 받음  
    def inner_func(*args, **kwargs):
        return func(*args, **kwargs)    
    return inner_func

• Example

def decorator(func):
    def wrapper(*args, **kwargs):       
        print('전처리')   
        print(func(*args, **kwargs))
        print('후처리')
    return wrapper
    
@decorator
def example():
    return '함수'
    
example()

# result
전처리
함수
후처리

• 클래스로 만들 때

class Decorator:

    def __init__(self, function):
        self.function = function

    def __call__(self, *args, **kwargs):
        return self.function(*args, **kwargs)

• Example

class Decorator:

    def __init__(self, function):
        self.function = function

    def __call__(self, *args, **kwargs):
        print('전처리')
        print(self.function(*args, **kwargs))
        print('후처리')
        
@Decorator
def example():
    return '클래스'

example()

# result
전처리
클래스
후처리