🧤 함수로 장식하자!

Decorator는 언제 쓸까?

데코레이터는 함수를 수정하지 않은 상태에서 추가 기능을 구현할 때 사용된다.

데코레이터 써보기

---

시나리오 |

고객 상담 센터 직원 김애플님은 회사 규정에 따라 전화를 걸거나 받을 때마다 본인 부서, 본인 이름을 밝힌 다음 본론으로 들어갈 수 있다.

데코레이터 없는 일반 코드

def greeting():
  print('안녕하세요 전자센터 고객 센터 1팀 소속 직원 김애플입니다.')

  print('박삼성 고객님께서 <노트북 수리> 스마트 다이얼 번호를 누르셔서 해당 부서로 연결됐습니다. 혹시 문제가 있으신가요?')
  
greeting()

데코레이터 없는 일반 코드 실행 결과

안녕하세요 전자센터 고객 센터 1팀 소속 직원 김애플입니다.
박삼성 고객님께서 <노트북 수리> 스마트 다이얼 번호를 누르셔서 해당 부서로 연결됐습니다. 혹시 문제가 있으신가요?

문제점: 박삼성님의 전화가 아닌 다른 고객의 전화, 또는 회사 동료한테 무엇을 질문할 때도 '안녕하세요 ~ 소속 ~ 김애플입니다.' 인사를 반복해야 한다. 인사를 매번 하기도 귀찮고! 사람은 실수하기 때문에 잊어버리고 안할 수도 있다! 간단한 데코레이터로 김애플님을 도와드릴 수 있다.

데코레이터 추가한 코드

def greeting(func):
  def wrapper():
    print('안녕하세요 전자센터 고객 센터 1팀 소속 직원 김애플입니다.')
    func() # 집중집중!
  return wrapper

def customer():
  customer_name = '박삼성'
  print(f'{customer_name} 고객님께서 <노트북 수리> 스마트 다이얼 번호를 누르셔서 해당 부서로 연결됐습니다. 혹시 문제가 있으신가요?')

def question():
  employee_name = '필립스'
  print(f'{employee_name}님 혹시 시간 괜찮으시다면 뭐 하나 여쭤봐도 될까요?')

greeting_customer = greeting(customer)
greeting_customer()

greeting_question = greeting(question)
greeting_question()

데코레이터 추가한 코드 실행 결과

안녕하세요 전자센터 고객 센터 1팀 소속 직원 김애플입니다.
박삼성 고객님께서 <노트북 수리> 스마트 다이얼 번호를 누르셔서 해당 부서로 연결됐습니다. 혹시 문제가 있으신가요?
안녕하세요 전자센터 고객 센터 1팀 소속 직원 김애플입니다.
필립스님 혹시 시간 괜찮으시다면 뭐 하나 여쭤봐도 될까요?

김애플님은 이제 인사지옥에서 빠져 나오셨다.

데코레이터의 구조:

1. trace : 데코레이터인 trace는 호출 함수를 매개변수로 받는다.
2. wrapper : trace 함수 안에서 호출 함수를 감싸는 함수다.

@ 골뱅이 데코레이터

---

@을 쓰면 더 간편하게 데코레이터를 구현할 수 있다.

기본 구조

@decorator
def new_function():
	...

@ 데코레이터 추가한 코드

def greeting(func):
  def wrapper():
    print('안녕하세요 전자센터 고객 센터 1팀 소속 직원 김애플입니다.')
    func()
  return wrapper

@greeting
def customer():
  customer_name = '박삼성'
  print(f'{customer_name} 고객님께서 <노트북 수리> 스마트 다이얼 번호를 누르셔서 해당 부서로 연결됐습니다. 혹시 문제가 있으신가요?')

@greeting
def question():
  employee_name = '필립스'
  print(f'{employee_name}님 혹시 시간 괜찮으시다면 뭐 하나 여쭤봐도 될까요?')

customer()
question()

@ 데코레이터 추가한 코드 실행 결과

안녕하세요 전자센터 고객 센터 1팀 소속 직원 김애플입니다.
박삼성 고객님께서 <노트북 수리> 스마트 다이얼 번호를 누르셔서 해당 부서로 연결됐습니다. 혹시 문제가 있으신가요?
안녕하세요 전자센터 고객 센터 1팀 소속 직원 김애플입니다.
필립스님 혹시 시간 괜찮으시다면 뭐 하나 여쭤봐도 될까요?

위에 trace() 썼던 방법과 똑같은 실행 결과가 나온 것을 확인할 수 있다~~

추가 | 데코레이터 여러개 지정하기

---

여러개의 데코레이터를 하나의 함수에 지정할 수 있다. 데코레이터는 위에서 아래로 실행된다!

시나리오 | 특별 이벤트 행사 기간이라 김애플님은 모든 상담 고객님께 행사 시간에 관한 정보도 추가로 전달해야 한다.

데코레이터 여러개 추가한 코드

def special_event(func):
  def wrapper():
    print('지금 특별 이벤트를 진행하고 있습니다. 상담 전화 후 전자센터 홈페이지를 방문하시면 더 자세한 정보를 확인하실 수 있습니다.')
    func()
  return wrapper

def greeting(func):
  def wrapper():
    print('안녕하세요 전자센터 고객 센터 1팀 소속 직원 김애플입니다.')
    func()
  return wrapper

@greeting
@special_event
def customer():
  customer_name = '박삼성'
  print(f'{customer_name} 고객님께서 <노트북 수리> 스마트 다이얼 번호를 누르셔서 해당 부서로 연결됐습니다. 혹시 문제가 있으신가요?')

customer()

데코레이터 여러개 추가한 코드 실행 결과

안녕하세요 전자센터 고객 센터 1팀 소속 직원 김애플입니다.
지금 특별 이벤트를 진행하고 있습니다. 상담 전화 후 전자센터 홈페이지를 방문하시면 더 자세한 정보를 확인하실 수 있습니다.
박삼성 고객님께서 <노트북 수리> 스마트 다이얼 번호를 누르셔서 해당 부서로 연결됐습니다. 혹시 문제가 있으신가요?

이벤트에 관한 사항도 @special_event 데코레이터를 활용해 성공적으로 전달했다!

Reference:
https://medium.com/@hckcksrl/python-%EB%8D%B0%EC%BD%94%EB%A0%88%EC%9D%B4%ED%84%B0-decorator-980fe8ca5276
https://dojang.io/mod/page/view.php?id=2427