• 정의

  • 다른 함수를 돌려주는 함수
  • 데코레이터는 편의 문법일 뿐이며 아래 두 함수 정의는 의미상 같다.

  def f(arg):
      ...
  f = staticmethod(f)
  
  @staticmethod
  def f(arg):
      ...
  

• 장점

  • 중복되는 코드 제거
  • 관심사의 분리: 함수 본인의 기능에만 집중하고 데코레이터가 나머지 역할을 맡도록 할 수 있습니다.

• 주의사항

  • 데코레이터 사용 시 함수의 메타데이터가 변경될 수 있습니다.
  • 여러 데코레이터를 사용할 때 위에 있는 데코레이터가 먼저 실행됩니다.

• 사용 사례

  • @classmethod, @staticmethod, @property
  • REST API에서 요청 핸들링
  • 이벤트 핸들링
  • getter, setter
  • 타이머
  • 로깅

• 예제

  def log(func):
      def wrapper(*args, **kwargs):
          print(f'start {func.__name__} function')
          
          # func는 아래의 printHello 함수입니다.
          func(*args, **kwargs)
          print(f'end {func.__name__} function')
      return wrapper# 장식할😊 함수를 반환
  
  @log
  def printHello():
      print("Hello")
  
  printHello()
  

  • 출력

  start printHello function
  Hello
  end printHello functiont
  

함수 호출을 제한하는 데코레이터도 만들 수 있다.

def limit_calls_decorator(max_calls):
    def decorator(func):
        calls = 0

        def wrapper(*args, **kwargs):
            nonlocal calls
            if calls < max_calls:
                calls += 1
                return func(*args, **kwargs)
            else:
                raise Exception("함수 호출 횟수 초과")

        return wrapper

    return decorator

아래 실행 결과를 보면 3번째 실행할 때까지는 문제가 없었지만 그 이후는 예외가 발생한다.

예제 함수 실행
예제 함수 실행
예제 함수 실행
함수 호출 횟수 초과
함수 호출 횟수 초과
Traceback (most recent call last):
...

클래스 안에서도 데코레이터를 사용할 수 있다.

class TestExample:
    def _decorator(func):
        def wrap(self, *args, **kargs):
            print("Start", func.__name__)
            func(self, *args, **kargs)
            print("End", func.__name__)

        return wrap

    @_decorator
    def test(self, a, b, c):
        print("Variables :", a, b, c)

t = TestExample()
t.test("1", 2, c="345")

클래스를 사용해서도 데코레이터를 만들 수 있다.

아래는 함수가 호출될 때마다 count변수를 증가시킨다. 함수의 호출 횟수를 알 수 있다는 말이다.

class emoticon:
		def __init__(self, emoticon):
				self.count = 0# 호출 회수 저장 필드self.emoticon = emoticon

		def __call__(self, func):
				def inner(*args, **kwargs):
						self.count += 1
						inner.count = self.count# 함수 객체에서 바로 사용 할 수 있도록
						print(self.emoticon, ' ', end='')
						return func(*args, **kwargs)

				return inner

@emoticon('-_-')
def say_hi(name):
		print(f'hi {name}!')

say_hi('Jason')
print(say_hi.count)