一、裝飾器介紹

1.1 為何要用裝飾器

軟件的設計應該遵循開放封閉原則，即對擴展是開放的，而對修改是封閉的：

為企業提供成都網站建設、成都網站制作、網站優化、網絡營銷推廣、競價托管、品牌運營等營銷獲客服務。成都創新互聯公司擁有網絡營銷運營團隊，以豐富的互聯網營銷經驗助力企業精準獲客，真正落地解決中小企業營銷獲客難題，做到“讓獲客更簡單”。自創立至今，成功用技術實力解決了企業“網站建設、網絡品牌塑造、網絡營銷”三大難題，同時降低了營銷成本，提高了有效客戶轉化率，獲得了眾多企業客戶的高度認可！

• 對擴展開放，意味著有新的需求或變化時，可以對現有代碼進行擴展，以適應新的情況。

• 對修改封閉，意味著對象一旦設計完成，就可以獨立完成其工作，而不要對其進行修改。

軟件包含的所有功能的源代碼以及調用方式，都應該避免修改，否則一旦改錯，則極有可能產生連鎖反應，最終導致程序崩潰。而對于上線后的軟件，新需求或者變化又層出不窮，我們必須為程序提供擴展的可能性，這就用到了裝飾器。

1.2 什么是裝飾器

‘裝飾’代指為被裝飾對象添加新的功能，‘器’代指器具/工具，裝飾器與被裝飾的對象均可以是任意可調用對象。

概括的講，裝飾器的作用就是在不修改被裝飾對象源代碼和調用方式的前提下為被裝飾對象添加額外的功能。

裝飾器經常用于有切面需求的場景，比如：插入日志、性能測試、事務處理、緩存、權限校驗等應用場景，裝飾器是解決這類問題的絕佳設計。有了裝飾器，就可以抽離出大量與函數功能本身無關的雷同代碼并繼續重用。

提示：可調用對象有函數，方法或者類，此處我們單以本篇主題：函數為例，來介紹函數裝飾器，并且被裝飾的對象也是函數。

二、裝飾器的實現

函數裝飾器分為：無參裝飾器和有參裝飾器兩種，二者的實現原理一樣，都是‘函數嵌套+閉包+函數對象’的組合使用的產物。

2.1 無參裝飾器的實現

如果想為下述函數添加統計其執行時間的功能：

import time
 
def index():
    time.sleep(3)
    print('Welcome to the index page')
    return 200
 
index()  # 函數執行

遵循不修改被裝飾對象源代碼的原則，我們想到的解決方法可能是這樣：

start_time = time.time()
index()  # 函數執行
stop_time = time.time()
print('run time is %s' % (stop_time - start_time))

考慮到還有可能要統計其他函數的執行時間，于是我們將其做成一個單獨的工具，函數體需要外部傳入被裝飾的函數從而進行調用，我們可以使用參數的形式傳入：

def wrapper(func):  # 通過參數接收外部的值
    start_time = time.time()
    res = func()
    stop_time = time.time()
    print('run time is %s' % (stop_time - start_time))
    return res

但之后函數的調用方式都需要統一改成：

wrapper(index)
wrapper(其他函數)

這便違反了不能修改被裝飾對象調用方式的原則，于是我們需要換一種為函數體傳值的方式，那之前所學的閉包函數在這里就能派上用場了，修改如下：

def timer(func):
    def wrapper():  # 引用外部作用域的變量func
        start_time = time.time()
        res = func()
        stop_time = time.time()
        print('run time is %s' % (stop_time - start_time))
        return res
    return wrapper

這樣我們便可以在不修改被裝飾函數源代碼和調用方式的前提下為其加上統計時間的功能，只不過需要事先執行一次timer將被裝飾的函數傳入，返回一個閉包函數wrapper重新賦值給變量名/函數名index，如下：

index = timer(index) 
index() 
"""
第1行得到index = wrapper，wrapper攜帶對外作用域的引用：func = 原始的index
第2行執行的是wrapper()，在wrapper的函數體內再執行最原始的index
"""

至此我們便實現了一個無參裝飾器timer，可以在不修改被裝飾對象index源代碼和調用方式的前提下為其加上新功能。但我們忽略了若被裝飾的函數是一個有參函數，便會拋出異常：

def home(name):
    time.sleep(5)
    print('Welcome to the home page', name)
 
home = timer(home)
home('jason')
# 拋出異常
TypeError: wrapper() takes 0 positional arguments but 1 was given 

之所以會拋出異常，是因為home(‘jason’)調用的其實是wrapper(‘jason’)，而函數wrapper沒有參數。wrapper函數接收的參數其實是給最原始的func用的，為了能滿足被裝飾函數參數的所有情況，便用上*args和**kwargs組合 ，于是修正后的裝飾器timer如下：

def timer(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        res = func(*args, **kwargs)
        stop_time = time.time()
        print('run time is %s' % (stop_time - start_time))
        return res
    return wrapper

此時我們就可以用timer來裝飾帶參數或不帶參數的函數了，但是為了簡潔而優雅地使用裝飾器，Python提供了專門的裝飾器語法糖來取代index = timer(index)的形式，需要在被裝飾對象的正上方單獨一行添加@timer（語法糖必須緊貼在被裝飾對象的上方）；

當解釋器解釋到@timer時就會調用timer函數，且把它正下方的函數名當做實參傳入，然后將返回的結果重新賦值給原函數名：

@timer    """index = timer(index)"""
def index():
    time.sleep(3)
    print('Welcome to the index page')
    return 200

@timer    """index = timer(home)"""
def home(name):
    time.sleep(5)
    print('Welcome to the home page', name)

裝飾器語法糖內部原理：會自動將下面緊貼著的被裝飾對象名字當做參數傳給裝飾器調用。

如果我們有多個裝飾器，可以疊加多個：

@deco3
@deco2
@deco1
def index():
    pass

疊加多個裝飾器也無特殊之處，上述代碼語義如下：

index = deco3(deco2(deco1(index)))

2.2 有參裝飾器的實現

了解無參裝飾器的實現原理后，我們可以再實現一個用來為被裝飾對象添加認證功能的裝飾器，實現的基本形式如下：

def deco(func):
    def wrapper(*args, **kwargs):
        編寫基于文件的認證,認證通過則執行res = func(*args,**kwargs),并返回res
    return wrapper

如果我們想提供多種不同的認證方式以供選擇，單從wrapper函數的實現角度改寫如下：

def deco(func):
	def wrapper(*args, **kwargs):
        if driver == 'file':
            編寫基于文件的認證,認證通過則執行res = func(*args, **kwargs),并返回res
        elif driver == 'mysql':
            編寫基于mysql認證,認證通過則執行res = func(*args, **kwargs),并返回res
	return wrapper

函數wrapper需要一個driver參數，而函數deco與wrapper的參數都有其特定的功能，不能用來接收其他類別的參數，既然需要一個參數，那我們可以再在deco的外部再包一層函數auth，用來專門接收額外的參數，這樣便保證了在auth函數內無論多少層都可以引用到：

def auth(driver):
    def deco(func):
        ……
    return deco

此時我們就實現了一個有參裝飾器，使用方式如下：

"""先調用auth(driver='file')，得到@deco，deco是一個閉包函數，包含了對外部作用域名字driver的引用，@deco的語法意義與無參裝飾器一樣"""
@auth(driver='file')
def index():
    pass
 
"""再調用auth(driver='mysql')，后續同上"""    
@auth(driver='mysql')
def home():
    pass

可以使用help(函數名)來查看函數的文檔注釋，本質就是查看函數的doc屬性，但對于被裝飾之后的函數，查看文檔注釋：

@timer
def home(name):
    '''
    home page function
    :param name: str
    :return: None
    '''
    time.sleep(5)
    print('Welcome to the home page', name)
    
print(help(home))
"""
打印結果：
 
Help on function wrapper in module __main__:
 
wrapper(*args, **kwargs)
 
None
"""

在被裝飾之后home = wrapper，查看home.name也可以發現home的函數名其實是wrapper，如果想要保留原函數的文檔和函數名屬性，就需要修正裝飾器：

def timer(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        res = func(*args, **kwargs)
        stop_time = time.time()
        print('run time is %s' % (stop_time - start_time))
        return res
    wrapper.__doc__ = func.__doc__
    wrapper.__name__ = func.__name__
    return wrapper

可以看到，按照上述方式來實現保留原函數屬性過于麻煩，functools模塊下提供一個裝飾器wraps專門用來幫我們實現這件事，用法如下：

from functools import wraps
 
def timer(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        start_time = time.time()
        res = func(*args, **kwargs)
        stop_time = time.time()
        print('run time is %s' % (stop_time - start_time))
        return res
    return wrapper