一、函數(shù)回顧

1、在python中函數(shù)是一等公民，函數(shù)也是對(duì)象。我們可以把函數(shù)賦予變量。

def make_cofe(type):
    print('獲得一杯 : {}'.format(type))
​
get_cofe = make_cofe
get_cofe('咖啡')
​
####輸出#####
獲得一杯 : 咖啡

這個(gè)例子中，我們把函數(shù)make_cofe 賦予了變量 get_cofe，這樣之后你調(diào)用 get_cofe，就相當(dāng)于是調(diào)用函數(shù) make_cofe()。

2、把函數(shù)當(dāng)作參數(shù)，傳入另一個(gè)函數(shù)中。

def make_cofe(type):
    print('獲得一杯 : {}'.format(type))
​
def shop(func,type):
    func(type)
​
shop(make_cofe,'咖啡')
​
####輸出####
獲得一杯 : 咖啡

這個(gè)例子，我們把make_cofe以參數(shù)的形式傳入shop中，然后調(diào)用它。

3、函數(shù)是可以嵌套的。

def shop(type):
    def make_cofe(type):
        print('獲得一杯 : {}'.format(type))
    make_cofe(type)
​
shop('咖啡')
​
#####輸出####
獲得一杯 : 咖啡

這段代碼中，我們?cè)诤瘮?shù)shop內(nèi)部定義了函數(shù)make_cofe 4、函數(shù)的返回值也可以是函數(shù)對(duì)象（閉包）。

def shop():
    def make_cofe(type):
        print('獲得一杯 : {}'.format(type))
    return make_cofe
​
get_cofe=shop()
get_cofe("咖啡")
​
####輸出#####
獲得一杯 : 咖啡

這里，函數(shù) shop() 的返回值是函數(shù)對(duì)象 make_cofe 本身，之后，我們將其賦予變量 get_cofe，再調(diào)用 get_cofe("咖啡")。

二、裝飾器

下面我們正式開(kāi)始裝飾器的學(xué)習(xí)。 我們先想一個(gè)問(wèn)題。如果我們?nèi)タХ鹊暌槐Х?，我們?yīng)該如何實(shí)現(xiàn)。你也許會(huì)這么寫(xiě)。

def cofe():
    print('咖啡', end='')
​
cofe()
​
####輸出####
咖啡

那我們現(xiàn)在想來(lái)一杯加糖咖啡，我們?cè)撊绾螌?xiě)呢？你也許會(huì)這么想，那還不簡(jiǎn)單，直接在cofe()函數(shù)里改不就好了。

def cofe():
    print('加糖咖啡', end='')
​
cofe()
​
####輸出####
加糖咖啡

那么問(wèn)題來(lái)了，如果我們現(xiàn)在不想喝加糖咖啡了，該怎么辦呢，總不能在cofe()函數(shù)里去掉吧。那如果有人想喝加糖咖啡、有人不想喝加糖咖啡如何是好，總不能寫(xiě)兩個(gè)cofe()函數(shù)吧。 那我們帶著問(wèn)題看一下下面這段代碼。

def add_sugar(func):
    def add():
        print('加糖',end='')
        func()
    return add
​
def cofe():
    print('咖啡',end='')
​
cofe = add_sugar(cofe)
​
print("獲得一杯",end='')
cofe()
​
####輸出#####
獲得一杯加糖咖啡

變量 cofe 指向了內(nèi)部函數(shù) add()，而內(nèi)部函數(shù) add() 中又會(huì)調(diào)用原函數(shù) cofe()，因此，最后調(diào)用 cofe() 時(shí)，就會(huì)先打印‘加糖'，然后輸出‘咖啡'。這里的函數(shù) add_sugar() 就是一個(gè)裝飾器，它把真正需要執(zhí)行的函數(shù)cofe()包裹在其中，并且改變了它的行為，但是原函數(shù) cofe() 不變。 下面我們來(lái)看一下更優(yōu)雅的寫(xiě)法。

def add_sugar(func):
    def add():
        print('加糖',end='')
        func()
    return add
​
@add_sugar
def cofe():
    print('咖啡',end='')
​
print("獲得一杯",end='')
cofe()
​
#####輸出#####
獲得一杯加糖咖啡

這里的@叫做語(yǔ)法糖， @add_sugar就相當(dāng)于前面的cofe = add_sugar(cofe)語(yǔ)句，只不過(guò)更加簡(jiǎn)潔。因此程序中建議用這種寫(xiě)法。 好了，讓我們來(lái)回顧下我們的問(wèn)題，如果有人想喝加糖咖啡、有人不想喝加糖咖啡如何是好。學(xué)了裝飾器那不就很簡(jiǎn)單了，如果要喝加糖咖啡，我們把加糖的裝飾器@add_sugar給加上不就好了，如果喝不加糖的，那就不加裝飾器，這樣我們就把這個(gè)問(wèn)題給完美解決掉了。在不改變函數(shù)內(nèi)部的前提了，給函數(shù)又添加了新的功能。 到目前為止，我們已經(jīng)把最簡(jiǎn)單的裝飾器學(xué)完了。下面我們?cè)诳紤]一個(gè)問(wèn)題，如果原函數(shù) cofe() 中，有參數(shù)需要傳遞給裝飾器怎么辦？一個(gè)簡(jiǎn)單的辦法，是可以在對(duì)應(yīng)的裝飾器函數(shù) add() 上，加上相應(yīng)的參數(shù)。

def add_sugar(func):
    def add(type):
        print('加糖',end='')
        func(type)
    return add
​
@add_sugar
def cofe(type):
    print('{}咖啡'.format(type),end='')
​
​
cofe("美式")
print()
cofe("拿鐵")
​
####輸出#####
加糖美式咖啡
加糖拿鐵咖啡

不過(guò)，新的問(wèn)題來(lái)了。如果我另外還有一個(gè)函數(shù)（奶茶函數(shù)），也需要使用 add_sugar() 裝飾器，但是這個(gè)新的函數(shù)有兩個(gè)參數(shù)，又該怎么辦呢？ 通常情況下，我們會(huì)把*args和 **kwargs，作為裝飾器內(nèi)部函數(shù) add() 的參數(shù)。*args和**kwargs，表示接受任意數(shù)量和類(lèi)型的參數(shù)，因此加糖裝飾器就可以寫(xiě)成下面的形式：

def add_sugar(func):
    def add(*args, **kwargs):
        print('加糖',end='')
        func(*args, **kwargs)
    return add
     
@add_sugar
def cofe(type):
    print('{}咖啡'.format(type),end='')
​
@add_sugar
def milk_tea(type,num):
    print('{}杯{}奶茶'.format(num,type), end='')
​
​
cofe("美式")
print()
milk_tea("xx牌子","4")
​
####輸出####
加糖美式咖啡
加糖4杯xx牌子奶茶

這樣我們的咖啡和奶茶都可以加糖了。 前面我們講的是函數(shù)的裝飾器，下面我們來(lái)講一下類(lèi)作為裝飾器。類(lèi)裝飾器主要依賴(lài)于函數(shù)__call__()，每當(dāng)你調(diào)用一個(gè)類(lèi)的實(shí)例時(shí)，函數(shù)__call__()就會(huì)被執(zhí)行一次。

​class Add_sugar:
    def __init__(self, func):
        self.func = func
        self.add_suger = "加糖"
​
    def __call__(self, *args, **kwargs):
        print(self.add_suger,end='')
        return self.func(*args, **kwargs)
​
@Add_sugar
def cofe():
    print("咖啡")
​
cofe()
​
####輸出#####
加糖咖啡

最后如果我們的咖啡既要加糖又要加冰，那我們?cè)撊绾巫瞿?？我們定義一個(gè)加冰的裝飾器就好了呀。

def add_sugar(func):
    def add():
        print('加糖',end='')
        func()
    return add
​
def add_ice(func):
    def add():
        print('加冰',end='')
        func()
    return add
​
​
@add_sugar
@add_ice
def cofe():
    print('咖啡',end='')
​
cofe()
​
####輸出####
加糖加冰咖啡

到此這篇關(guān)于Python裝飾器-給你的咖啡加點(diǎn)料的文章就介紹到這了,更多相關(guān)Python裝飾器內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！