【Python】解析Python中的裝飾器

python中的函數也是對象，函數能夠被看成變量傳遞。

裝飾器在python中功能很是強大，裝飾器容許對原有函數行爲進行擴展，而不用硬編碼的方式，它提供了一種面向切面的訪問方式。

 

裝飾器

一個普通的裝飾器通常是這樣：

import functools
def log(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        print('call %s():' % func.__name__)
        print('args = {}'.format(*args))
        return func(*args, **kwargs)
    return wrapper

這樣就定義了一個打印出方法名及其參數的裝飾器。
調用之：

@log
def test(p):
    print(test.__name__ + " param: " + p)

test("I'm a param")

輸出：

call test():
args = I'm a param
test param: I'm a param

裝飾器在使用時，用了@語法，讓人有些困擾。其實，裝飾器只是個方法，與下面的調用方式沒有區別：

def test(p):
    print(test.__name__ + " param: " + p)

wrapper = log(test)
wrapper("I'm a param")

@語法只是將函數傳入裝飾器函數，並沒有神奇之處。
值得注意的是@functools.wraps(func)，這是python提供的裝飾器。它能把原函數的元信息拷貝到裝飾器裏面的 func 函數中。函數的元信息包括docstring、name、參數列表等等。能夠嘗試去除@functools.wraps(func)，你會發現test.__name__的輸出變成了wrapper。

 

帶參數的裝飾器

裝飾器容許傳入參數，一個攜帶了參數的裝飾器將有三層函數，以下所示：

import functools

def log_with_param(text):
    def decorator(func):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            print('call %s():' % func.__name__)
            print('args = {}'.format(*args))
            print('log_param = {}'.format(text))
            return func(*args, **kwargs)

        return wrapper

    return decorator
    
@log_with_param("param")
def test_with_param(p):
    print(test_with_param.__name__)

看到這個代碼是否是又有些疑問，內層的decorator函數的參數func是怎麼傳進去的？和上面通常的裝飾器不大同樣啊。
其實道理是同樣的，將其@語法去除，恢復函數調用的形式一看就明白了：

# 傳入裝飾器的參數，並接收返回的decorator函數
decorator = log_with_param("param")
# 傳入test_with_param函數
wrapper = decorator(test_with_param)
# 調用裝飾器函數
wrapper("I'm a param")

輸出結果與正常使用裝飾器相同：

call test_with_param():
args = I'm a param
log_param = param
test_with_param

至此，裝飾器這個有點費解的特性也沒什麼神祕了。
裝飾器這一語法體現了Python中函數是第一公民，函數是對象、是變量，能夠做爲參數、能夠是返回值，很是的靈活與強大。

原文連接：

https://www.jianshu.com/p/ee82b941772a
https://www.geeksforgeeks.org/decorators-in-python/