一、什么是裝飾器

所謂的裝飾器，其實就是通過裝飾器函數(shù)，來修改原函數(shù)的一些功能，使得原函數(shù)不需要修改。

這一句話理解起來可能沒那么輕松，那先來看一個"傻瓜"函數(shù)。

放心，絕對不是"Hello World"！

def hello():print("你好，裝飾器")

腫么樣，木騙你吧？ 哈哈，這個函數(shù)不用運行相信大家都知道輸出結果：“你好，裝飾器”。

那如果我想讓hello()函數(shù)再實現(xiàn)個其他功能，比如多打印一句話。

那么，可以這樣"增強"一下：

def my_decorator(func):def wrapper():print("這是裝飾后具有的新輸出")func()return wrapperdef hello():print("你好，裝飾器")hello = my_decorator(hello)hello()

運行結果：

這是裝飾后具有的新輸出 你好，裝飾器 [Finished in 0.1s]

很顯然，這個"增強"沒啥作用，但是可以幫助理解裝飾器。

當運行最后的hello()函數(shù)時，調(diào)用過程是這樣的：

hello = my_decorator(hello)中，變量hello指向的是my_decorator()

my_decorator(func)中傳參是hello，返回的wrapper,因此又會調(diào)用到原函數(shù)hello()

于是乎，先打印出了wrapper()函數(shù)里的，然后才打印出hello()函數(shù)里的

那上述代碼里的my_decorator()就是一個裝飾器。

它改變了hello()的行為，但是并沒有去真正的改變hello()函數(shù)的內(nèi)部實現(xiàn)。

但是，python一直以"優(yōu)雅"被人追捧，而上述的代碼顯然不夠優(yōu)雅。

二、優(yōu)雅的裝飾器

所以，想讓上述裝飾器變得優(yōu)雅，可以這樣寫：

''' 遇到問題沒人解答？小編創(chuàng)建了一個Python學習交流QQ群：778463939 尋找有志同道合的小伙伴，互幫互助,群里還有不錯的視頻學習教程和PDF電子書！ ''' def my_decorator(func):def wrapper():print("這是裝飾后具有的新輸出")func()return wrapper@my_decorator def hello():print("你好，裝飾器")hello()

這里的@my_decorator就相當于舊代碼的hello = my_decorator(hello)，@符號稱為語法糖。

那如果還有其他函數(shù)也需要加上類似的裝飾，直接在函數(shù)的上方加上@my_decorator就可以，大大提高函數(shù)的重復利用與可讀性。

''' 遇到問題沒人解答？小編創(chuàng)建了一個Python學習交流QQ群：778463939 尋找有志同道合的小伙伴，互幫互助,群里還有不錯的視頻學習教程和PDF電子書！ ''' def my_decorator(func):def wrapper():print("這是裝飾后具有的新輸出")func()return wrapper@my_decorator def hello():print("你好，裝飾器")@my_decorator def hello2():print("你好，裝飾器2")hello2()

輸出：

這是裝飾后具有的新輸出 你好，裝飾器2 [Finished in 0.1s]

三、帶參數(shù)的裝飾器

單個參數(shù)

上面的只是一個非常簡單的裝飾器，但是實際場景中，很多函數(shù)都是要帶有參數(shù)的，比如hello(people_name)。

其實也很簡單，要什么我們就給什么唄，直接在對應裝飾器的wrapper()上，加上對應的參數(shù)：

def my_decorator(func):def wrapper(people_name):print("這是裝飾后具有的新輸出")func(people_name)return wrapper@my_decorator def hello(people_name):print("你好，{}".format(people_name))hello("張三")

輸出：

這是裝飾后具有的新輸出 你好，張三 [Finished in 0.1s]

多個參數(shù)

但是還沒完，這樣雖然簡單，但是隨之而來另一個問題：因為并不是所有函數(shù)參數(shù)都是一樣的，
當其他要使用裝飾器的函數(shù)參數(shù)不止這個一個腫么辦？比如：

@my_decorator def hello3(speaker, listener):print("{}對{}說你好！".format(speaker, listener))

沒關系，在python里，*args和**kwargs表示接受任意數(shù)量和類型的參數(shù)，所以我們可以這樣
寫裝飾器里的wrapper()函數(shù)：

''' 遇到問題沒人解答？小編創(chuàng)建了一個Python學習交流QQ群：778463939 尋找有志同道合的小伙伴，互幫互助,群里還有不錯的視頻學習教程和PDF電子書！ ''' def my_decorator(func):def wrapper(*args, **kwargs):print("這是裝飾后具有的新輸出")func(*args, **kwargs)return wrapper@my_decorator def hello(people_name):print("你好，{}".format(people_name))@my_decorator def hello3(speaker, listener):print("{}對{}說你好！".format(speaker, listener))hello("老王") print("------------------------") hello3("張三", "李四")

同時運行下hello(“老王”)，和hello3(“張三”, “李四”)，看結果：

這是裝飾后具有的新輸出 你好，老王 ------------------------ 這是裝飾后具有的新輸出 張三對李四說你好！ [Finished in 0.1s]

自定義參數(shù)

上面2種，裝飾器都是接收外來的參數(shù)，其實裝飾器還可以接收自己的參數(shù)。
比如，我加個參數(shù)來控制下裝飾器中打印信息的次數(shù)：

''' 遇到問題沒人解答？小編創(chuàng)建了一個Python學習交流QQ群：778463939 尋找有志同道合的小伙伴，互幫互助,群里還有不錯的視頻學習教程和PDF電子書！ ''' def count(num):def my_decorator(func):def wrapper(*args, **kwargs):for i in range(num):print("這是裝飾后具有的新輸出")func(*args, **kwargs)return wrapperreturn my_decorator@count(3) def hello(people_name):print("你好，{}".format(people_name))hello("老王")

注意，這里count裝飾函數(shù)中的2個return.
運行下，應該會出現(xiàn)3次：

這是裝飾后具有的新輸出 你好，老王 這是裝飾后具有的新輸出 你好，老王 這是裝飾后具有的新輸出 你好，老王 [Finished in 0.1s]

內(nèi)置裝飾器@functools.wrap

現(xiàn)在多做一步探索，我們來打印下下面例子中的hello()函數(shù)的元信息：

''' 遇到問題沒人解答？小編創(chuàng)建了一個Python學習交流QQ群：778463939 尋找有志同道合的小伙伴，互幫互助,群里還有不錯的視頻學習教程和PDF電子書！ ''' def my_decorator(func):def wrapper(*args, **kwargs):print("這是裝飾后具有的新輸出")func(*args, **kwargs)return wrapper@my_decorator def hello(people_name):print("你好，{}".format(people_name))print(hello.__name__) #看下hello函數(shù)的元信息

輸出：

wrapper

這說明了，它不再是以前的那個 hello() 函數(shù)，而是被 wrapper() 函數(shù)取代了。

如果我們需要用到元函數(shù)信息，那怎么保留它呢？這時候可以用內(nèi)置裝飾器@functools.wrap。

import functoolsdef my_decorator(func):@functools.wraps(func)def wrapper(*args, **kwargs):print("這是裝飾后具有的新輸出")func(*args, **kwargs)return wrapper@my_decorator def hello(people_name):print("你好，{}".format(people_name))print(hello.__name__)

運行下：

hello [Finished in 0.1s]

四、類裝飾器

裝飾器除了是函數(shù)之外，也可以是類。

但是類作為裝飾器的話，需要依賴一個函數(shù)__call__()，當調(diào)用這個類的實例時，函數(shù)__call__()就
會被執(zhí)行。

來改造下之前的例子，把函數(shù)裝飾器改成類裝飾器：

''' 遇到問題沒人解答？小編創(chuàng)建了一個Python學習交流QQ群：778463939 尋找有志同道合的小伙伴，互幫互助,群里還有不錯的視頻學習教程和PDF電子書！ ''' class MyDecorator():def __init__(self, func):self.func = funcdef __call__(self, *args, **kwargs):print("這是裝飾后具有的新輸出")return self.func(*args, **kwargs)# def my_decorator(func): # def wrapper(): # print("這是裝飾后具有的新輸出") # func() # return wrapper@MyDecorator def hello():print("你好，裝飾器")hello()

運行：

這是裝飾后具有的新輸出 你好，裝飾器 [Finished in 0.1s]

跟函數(shù)裝飾器一樣，實現(xiàn)一樣的功能。

五、裝飾器的嵌套

既然裝飾器可以增強函數(shù)的功能，那如果有多個裝飾器，我都想要怎么辦？
其實，只要把需要用的裝飾器都加上去就好了：

@decorator1 @decorator2 @decorator3 def hello():...

但是要注意這里的執(zhí)行順序，會從上到下去執(zhí)行，可以來看下：

''' 遇到問題沒人解答？小編創(chuàng)建了一個Python學習交流QQ群：778463939 尋找有志同道合的小伙伴，互幫互助,群里還有不錯的視頻學習教程和PDF電子書！ ''' def my_decorator(func):def wrapper():print("這是裝飾后具有的新輸出")func()return wrapperdef my_decorator2(func):def wrapper():print("這是裝飾后具有的新輸出2")func()return wrapperdef my_decorator3(func):def wrapper():print("這是裝飾后具有的新輸出3")func()return wrapper@my_decorator @my_decorator2 @my_decorator3 def hello():print("你好，裝飾器")hello()

運行

這是裝飾后具有的新輸出 這是裝飾后具有的新輸出2 這是裝飾后具有的新輸出3 你好，裝飾器 [Finished in 0.1s]

好記性不如爛筆頭，寫一下理解一下會好很多。

總結

以上是生活随笔為你收集整理的一篇文章带你了解python装饰器的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。