Decorator_02 (파라미터가 있는 함수 decorator & 모든 함수에 대한 decorator)

Decorator

- 기존 함수에 기능을 추가하여 새로운 함수를 만듬

- Closure function을 활용

- https://www.python.org/dev/peps/pep-0318/

- decorator 사용 이유는? 함수 혹은 메쏘드에 새로운 기능을 추가하기 위해!

- 그렇다면 왜 소스 코드를 수정하지 않는가? 여러 함수에 동일한 기능을 추가할 수 있다.

  예를 들면, 모든 함수에 전달된 파라미터의 유효성 검사가 필요하다고 가정했을때, 유효성 검사 코드가 각 함수마다 복사되면 수정의 어려움이 존재한다.

 

# decorator 함수 정의
def decorate_func(function):
    def inner():
        print('decoration added')
        function()
    return inner
 
def simple():
    print('simpel function')
    
# 일반 함수 호출
simple()
 
print('='*20)
 
# simple 함수에 기능을 추가한(decorate 한) decorated 함수
decorated = decorate_func(simple)
# 결과가 decoration 됨
decorated()
 
==================<<실행결과>>==================
 
simpel function
====================
decoration added
simpel function
 

 

 

'@'심볼 사용

- decorator를 생성하기 위한 syntactic sugar (문법적 편의성)

# decorator 함수 정의
def decorate_func(function):
    def inner():
        print('decoration added')
        function()
    return inner
 
def simple():
    print('simple function')
    
# 사실 귀찮고, python 답지 않은 문법임
simple = decorate_func(simple)
simple()
 
==================<<실행결과>>==================
 
decoration added
simple function

 

# decorator 함수 정의
def decorate_func(fn):
    def inner():
        print('decoration added')
        fn()
    return inner
 
# 아래와 같이 @를 사용하여 decoration 가능!
# 기능을 추가하는 것이라고 이해하면 된다.
@decorate_func
def simple():
    print('simple function')
    
# 결과가 decorated 됨!
simple()
 
==================<<실행결과>>==================
 
decoration added
simple function

 

 

파라미터가 있는 함수 Decorator

- 중첩함수에 꾸미고자 하는 함수와 동일하게 파라미터를 가져가면 됨

 

case1) '@' 기호를 쓰지 않은 경우

def decorate_divide(fn):
    def wrapper(a, b):
        if b == 0:
            print('zero cannot be divided!')
            return
        return fn(a, b)
    return wrapper
 
def divide(a, b):
    return a / b
 
divide = decorate_divide(divide)
print(divide(9, 3))
print(divide(9, 0))
 
==================<<실행결과>>==================
 
3.0
zero cannot be divided!
None