클래스에서 메서드를 만들 때 @staticmethod, @classmethod, @abstractmethod 등을 붙였는데, 이렇게 @로 시작하는 것들이 데코레이터. - 함수(메서드)를 장식
함수를 수정하지 않은 상태에서 추가 기능을 구현할 때 사용
예)
def hello():
print('hello 함수 시작')
print('hello')
print('hello 함수 끝')
def world():
print('world 함수 시작')
print('world')
print('world 함수 끝')
hello()
world()
결과:
hello 함수 시작
hello
hello 함수 끝
world 함수 시작
world
world 함수 끝
다른 함수에도 적용할 때 번거로움 -> 데코레이터 사용
def trace(func): # 호출할 함수를 매개변수로 받음
def wrapper(): # 호출할 함수를 감싸는 함수
print(func.name, '함수 시작') # name으로 함수 이름 출력
func() # 매개변수로 받은 함수를 호출
print(func.name, '함수 끝')
return wrapper # wrapper 함수 반환
def hello():
print('hello')
def world():
print('world')
trace_hello = trace(hello) # 데코레이터에 호출할 함수를 넣음
trace_hello() # 반환된 함수를 호출
trace_world = trace(world) # 데코레이터에 호출할 함수를 넣음
trace_world() # 반환된 함수를 호출
result:
hello 함수 시작
hello
hello 함수 끝
world 함수 시작
world
world 함수 끝
데코레이터 trace는 호출할 함수를 매개변수로 받음
trace 함수 안에서는 호출할 함수를 감싸는 함수 wrappe
wrapper 함수에서는 함수의 시작을 알리는 문자열을 출력하고, trace에서 매개변수로 받은 func를 호출, 함수의 끝을 알리는 문자열을 출력
매개변수로 받은 함수의 원래 이름을 출력할 때는 name 속성을 활용
wrapper 함수를 다 만들었으면 return을 사용하여 wrapper 함수 자체를 반환
-> 함수 안에서 함수를 만들고 반환하는 클로저
데코레이터를 사용할 때는 trace에 호출할 함수 hello 또는 world를 넣습니다. 그다음에 데코레이터에서 반환된 함수를 호출
@로 데코레이터 사용
@데코레이터1
@데코레이터2
def 함수이름():
코드
예)
hello와 world 함수 위에 @trace를 붙인 뒤에 hello와 world 함수를 그대로 호출하면 끝
def trace(func): # 호출할 함수를 매개변수로 받음
def wrapper():
print(func.name, '함수 시작') # name으로 함수 이름 출력
func() # 매개변수로 받은 함수를 호출
print(func.name, '함수 끝')
return wrapper # wrapper 함수 반환
@trace # @데코레이터
def hello():
print('hello')
@trace # @데코레이터
def world():
print('world')
hello() # 함수를 그대로 호출
world() # 함수를 그대로 호출
