詳解Python裝飾器 給你的咖啡加點料

Posted on by

 一、函數回顧

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

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

這個例子中，我們把函數make_cofe 賦予瞭變量 get_cofe，這樣之後你調用 get_cofe，就相當於是調用函數 make_cofe()。

2、把函數當作參數，傳入另一個函數中。

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

這個例子，我們把make_cofe以參數的形式傳入shop中，然後調用它。

3、函數是可以嵌套的。

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

這段代碼中，我們在函數shop內部定義瞭函數make_cofe 4、函數的返回值也可以是函數對象（閉包）。

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

這裡，函數 shop() 的返回值是函數對象 make_cofe 本身，之後，我們將其賦予變量 get_cofe，再調用 get_cofe(“咖啡”)。

二、裝飾器

下面我們正式開始裝飾器的學習。 我們先想一個問題。如果我們去咖啡店要一杯咖啡，我們應該如何實現。你也許會這麼寫。

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

那我們現在想來一杯加糖咖啡，我們該如何寫呢？你也許會這麼想，那還不簡單，直接在cofe()函數裡改不就好瞭。

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

那麼問題來瞭，如果我們現在不想喝加糖咖啡瞭，該怎麼辦呢，總不能在cofe()函數裡去掉吧。那如果有人想喝加糖咖啡、有人不想喝加糖咖啡如何是好，總不能寫兩個cofe()函數吧。 那我們帶著問題看一下下面這段代碼。

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

變量 cofe 指向瞭內部函數 add()，而內部函數 add() 中又會調用原函數 cofe()，因此，最後調用 cofe() 時，就會先打印‘加糖’，然後輸出‘咖啡’。這裡的函數 add_sugar() 就是一個裝飾器，它把真正需要執行的函數cofe()包裹在其中，並且改變瞭它的行為，但是原函數 cofe() 不變。 下面我們來看一下更優雅的寫法。

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

這裡的@叫做語法糖， @add_sugar就相當於前面的cofe = add_sugar(cofe)語句，隻不過更加簡潔。因此程序中建議用這種寫法。 好瞭，讓我們來回顧下我們的問題，如果有人想喝加糖咖啡、有人不想喝加糖咖啡如何是好。學瞭裝飾器那不就很簡單瞭，如果要喝加糖咖啡，我們把加糖的裝飾器@add_sugar給加上不就好瞭，如果喝不加糖的，那就不加裝飾器，這樣我們就把這個問題給完美解決掉瞭。在不改變函數內部的前提瞭，給函數又添加瞭新的功能。 到目前為止，我們已經把最簡單的裝飾器學完瞭。下面我們在考慮一個問題，如果原函數 cofe() 中，有參數需要傳遞給裝飾器怎麼辦？一個簡單的辦法，是可以在對應的裝飾器函數 add() 上，加上相應的參數。

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("拿鐵")
​
####輸出#####
加糖美式咖啡
加糖拿鐵咖啡

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

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牌子奶茶

這樣我們的咖啡和奶茶都可以加糖瞭。 前面我們講的是函數的裝飾器，下面我們來講一下類作為裝飾器。類裝飾器主要依賴於函數__call__()，每當你調用一個類的實例時，函數__call__()就會被執行一次。

​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()
​
####輸出#####
加糖咖啡

最後如果我們的咖啡既要加糖又要加冰，那我們該如何做呢？我們定義一個加冰的裝飾器就好瞭呀。

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()
​
####輸出####
加糖加冰咖啡

到此這篇關於Python裝飾器-給你的咖啡加點料的文章就介紹到這瞭,更多相關Python裝飾器內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：