裝飾器函數(shù)是在不改變?cè)瓉?lái)函數(shù)的情況下，添加額外功能的函數(shù)。裝飾器函數(shù)內(nèi)部還有一個(gè)函數(shù)，返回值是內(nèi)部函數(shù)名。定義裝飾器函數(shù)的語(yǔ)法格式如下：

def 裝飾器函數(shù)名(參數(shù)1):

函數(shù)體

def 內(nèi)部函數(shù)名(參數(shù)2):

函數(shù)體

return 內(nèi)部函數(shù)名

其中，參數(shù)1接收原函數(shù)的函數(shù)名;參數(shù)2接收原函數(shù)的參數(shù)。

例如，某月餅零售價(jià)為5元一個(gè)，但是禮盒裝月餅(8個(gè)裝)為200元一盒，那么同樣的月餅，使用禮盒包裝后，可能就加價(jià)160元。

首先定義兩個(gè)函數(shù)，一個(gè)是月餅函數(shù)，另一個(gè)是禮盒函數(shù)。

定義月餅函數(shù)，代碼如下：

def cake():

print('五仁月餅零售價(jià)5元')

調(diào)用月餅函數(shù)，代碼如下：

cake()

調(diào)用月餅函數(shù)后，運(yùn)行結(jié)果：

五仁月餅零售價(jià)5元

定義禮盒函數(shù)，代碼如下：

def box():

print('禮盒加價(jià)160元')

def get_cake():

cake()

調(diào)用禮盒函數(shù)，代碼如下：

box()

調(diào)用禮盒函數(shù)后，運(yùn)行結(jié)果：

禮盒加價(jià)160元

運(yùn)行結(jié)果顯然不符合預(yù)期，沒(méi)有打印“五仁月餅零售價(jià)5元”，這是因?yàn)橹皇窃诙Y盒函數(shù)內(nèi)定義了get_cake()函數(shù)，但是沒(méi)有調(diào)用這個(gè)函數(shù)。修改禮盒函數(shù)，代碼如下：

def box():

print('禮盒加價(jià)160元')

def get_cake():

cake()

get_cake() # 調(diào)用函數(shù)

再次調(diào)用禮盒函數(shù)后，運(yùn)行結(jié)果：

禮盒加價(jià)160元

五仁月餅零售價(jià)5元

但是，同一種禮盒可能會(huì)裝不同的月餅，現(xiàn)有三種月餅分別為五仁、豆沙、蛋黃，這三種月餅的零售價(jià)分別為5元、5元、8元。

定義豆沙月餅函數(shù)，代碼如下：

def cake1():

print('豆沙月餅零售價(jià)5元')

定義蛋黃月餅函數(shù)，代碼如下：

def cake2():

print('蛋黃月餅零售價(jià)8元')

下面分別給禮盒函數(shù)傳遞三個(gè)原函數(shù)的名稱，修改禮盒函數(shù)，代碼如下：

def box(fun): # 添加參數(shù)

print('禮盒加價(jià)160元')

def get_cake():

fun()

get_cake()

這樣就分別實(shí)現(xiàn)了三種打印效果，三種月餅都可以被禮盒包裝了。

重新調(diào)用禮盒函數(shù)，代碼如下：

box(cake)

box(cake1)

box(cake2)

調(diào)用禮盒函數(shù)后，運(yùn)行結(jié)果：

禮盒加價(jià)160元

五仁月餅零售價(jià)5元

禮盒加價(jià)160元

豆沙月餅零售價(jià)5元

禮盒加價(jià)160元

蛋黃月餅零售價(jià)8元

目前月餅函數(shù)是無(wú)參數(shù)、無(wú)返回值的。給月餅函數(shù)添加返回值，代表月餅的個(gè)數(shù)，代碼如下：

def cake():

print('五仁月餅零售價(jià)5元')

return 4 # 添加返回值

而禮盒函數(shù)也需要進(jìn)行相應(yīng)的修改，代碼如下：

def box(fun):

print('禮盒加價(jià)160元')

def get_cake():

num = fun()

return num

return get_cake()

調(diào)用禮盒函數(shù)，代碼如下：

nums = box(cake)

print('獲取數(shù)量:%d' % nums)

調(diào)用禮盒函數(shù)后，運(yùn)行結(jié)果：

禮盒加價(jià)160元

五仁月餅零售價(jià)5元

獲取數(shù)量:4

目前原函數(shù)是cake()，其裝飾器函數(shù)是box()。Python中，原函數(shù)和裝飾器函數(shù)的修飾關(guān)系類(lèi)似于月餅和禮盒的關(guān)系。在定義原函數(shù)之前，使用“@+裝飾器函數(shù)名”的形式得到裝飾關(guān)系。

給月餅函數(shù)添加修飾關(guān)系，代碼如下：

@box

def cake():

print('五仁月餅零售價(jià)5元')

return 4

而禮盒函數(shù)也需要進(jìn)行相應(yīng)的修改，代碼如下：

def box(fun):

print('禮盒加價(jià)160元')

def get_cake():

num = fun()

return num

return get_cake # 返回值中刪除函數(shù)名后的括號(hào)

在調(diào)用時(shí)不需要再寫(xiě)禮盒函數(shù)名，就像最終吃的是月餅，因此，直接調(diào)用月餅函數(shù)即可，代碼如下：

nums = cake()

print('獲取數(shù)量:%d' % nums)

調(diào)用月餅函數(shù)后，運(yùn)行結(jié)果：

禮盒加價(jià)160元

五仁月餅零售價(jià)5元

獲取數(shù)量:4

由運(yùn)行結(jié)果可知，此時(shí)不需要寫(xiě)修飾器函數(shù)名，也可達(dá)到同樣的效果，這說(shuō)明了裝飾器的作用。

常見(jiàn)的函數(shù)一般是有參數(shù)的，裝飾器函數(shù)的參數(shù)用于接收原函數(shù)名，內(nèi)部函數(shù)的參數(shù)用于接收原函數(shù)的參數(shù)。有時(shí)原函數(shù)中參數(shù)的數(shù)量和類(lèi)型是無(wú)法事先確定的，可以使用*args和**kwargs，即可變位置參數(shù)和可變關(guān)鍵字參數(shù)，這樣無(wú)論任何參數(shù)都可以接收。

修改月餅函數(shù)，添加參數(shù)n，代表月餅的個(gè)數(shù)，代碼如下：

@box

def cake(n): # 添加參數(shù)

print('五仁月餅零售價(jià)5元')

return n # 修改返回值

而禮盒函數(shù)也需要進(jìn)行相應(yīng)的修改，代碼如下：

def box(fun):

print('禮盒加價(jià)160元')

def get_cake(*args, **kwargs): # 內(nèi)部函數(shù)添加參數(shù)

num = fun(*args, **kwargs)

return num

return get_cake

再次調(diào)用月餅函數(shù)，代碼如下：

nums = cake(4)

print('獲取數(shù)量:%d' % nums)

再次調(diào)用月餅函數(shù)后，運(yùn)行結(jié)果：

禮盒加價(jià)160元

五仁月餅零售價(jià)5元

獲取數(shù)量:4