[Python] 데코레이터 (Decorator)

데코레이터 (Decorator) ?

- 함수 앞뒤에 기능을 추가해서 손 쉽게 함수를 활용할 수 있는 기법
- Closure function
- 여러 함수에 동일한 기능을 @데코레이터 하나로 간편하게 추가

 

 

Closure function 

• 함수와 해당 함수가 가지고 있는 데이터를 함께 복사, 저장해서 별도 함수로 활용하는 기법
• 외부 함수가 소멸하더라도, 외부 함수 안에 있는 로컬 변수 값과 중첩함수(내부함수)를 사용 가능

def outer_func(num):
  #중첩 함수에서 외부 함수의 변수에 접근 가능
  def inner_func():
    print(num)
    return 'hi'
    
  return inner_func #중첩(내부) 함수 이름을 리턴
  
closure_func = outer_func(10) #First-class function
closure_func() #<---Closure 호출

#10
#'hi'

• 예제 코드 분석
  • 위의 예제에서 closure_func이 바로 closure
  • closure_func = outer_func(10)에서 outer_func함수는 호출 종료
  • closure_func()은 결국 inner_func 함수를 호출
  • outer_func(10) 호출 종료 시 num 값은 없어졌으나, closure_func()에서 inner_func이 호출되면서 이전의 num 값 10을 사용함
  • del outer_func ===> outer_func 함수를 아예 삭제해도, inner_func()와 num의 값 10은 살아있다!

 

 

데코레이터 작성법

#데코레이터 작성
def test_decorator(func): #func : 함수 안에 넣을 함수
  def wrapper():  #호출할 함수를 감싸는 함수
    print('함수 호출 전')  #함수 앞에서 실행할 내용
    func() #함수
    print('함수 호출 후') #함수 뒤에서 실행할 내용
    
  return wrapper #closure 함수로 만든다
  

#데코레이터 적용
@test_decorator
def logger_check():
  print("ok")
  
logger_check()

'''
함수 호출 전
ok
함수 호출 후
'''

 

• closure function 으로 데코레이터 풀어보기

def test_decorator(func):
  def wrapper(): 
    print('함수 호출 전') 
    func() #함수
    print('함수 호출 후') 
    
  return wrapper 
  
def logger_check():
	print('ok')

decorated_func = test_decorator(logger_check) 
decorated_func()

#@decorator 사용과 동일한 결과
'''
함수 호출 전
ok
함수 호출 후
'''

 

• 파라미터가 있는 함수에 Decorator 적용

#유효성 검사 decorator
def type_checker(function):
  def inner_func(digit1, digit2):
    if (type(digit1) != int) or (type(digit2) != int):
      print('only integer support')
      return
    return function(digit1, digit2)
  return inner_func
  
@type_checker
def multiplexer(digit1, digi2):
  return digit1 * digit2
  
muliplexer(1.1, 2) #only integer support

 

• 파라미터와 관계없이 모든 함수에 적용 가능한 Decorator 
  • 파라미터는 어떤 형태이든 결국 (args, **kwargs)로 표현 가능
  • 내부함수 파라미터에 적용 => 어떤 함수이든 데코레이터 적용 가능

#데코레이터 작성
def general_decorator(function):
  def wraaper(*args, **kwargs):
    print('function is decorated')
    return function(*args, **kwargs)
  return wrapper

#데코레이터 적용
@general_decorator
def calc_square(digit):
  return digit * digit
  
@general_decorator
def calc_plus(digit1, digit2):
  return digit1 + digit2
  
@general_decorator
def calc_quad(digit1, digit2, digit3, digit4):
  return digit1 * digit2 * digit3 * digit4

 

• 데코레이터 여러 개 지정

def decorator1(function)
  def wrapper():
    print('decorator1')
    funciton()
  return wrapper
  
def decorator2(function):
  def wrapper():
    print('decorator2')
    funciton()
  return wrapper
  

 #데코레이터 나열 순으로 실행 됨
@decorator1
@decorator2
def hello():
  print('hello')
  
"""
decorator1
decorator2
hello
"""

 

 

#참고

https://www.fun-coding.org/post/PL&OOP4-4.html#gsc.tab=0