一篇文章带你了解python装饰器
一、什么是裝飾器
所謂的裝飾器,其實就是通過裝飾器函數,來修改原函數的一些功能,使得原函數不需要修改。
這一句話理解起來可能沒那么輕松,那先來看一個"傻瓜"函數。
放心,絕對不是"Hello World"!
def hello():print("你好,裝飾器")腫么樣,木騙你吧? 哈哈,這個函數不用運行相信大家都知道輸出結果:“你好,裝飾器”。
那如果我想讓hello()函數再實現個其他功能,比如多打印一句話。
那么,可以這樣"增強"一下:
def my_decorator(func):def wrapper():print("這是裝飾后具有的新輸出")func()return wrapperdef hello():print("你好,裝飾器")hello = my_decorator(hello)hello()運行結果:
這是裝飾后具有的新輸出 你好,裝飾器 [Finished in 0.1s]很顯然,這個"增強"沒啥作用,但是可以幫助理解裝飾器。
當運行最后的hello()函數時,調用過程是這樣的:
那上述代碼里的my_decorator()就是一個裝飾器。
它改變了hello()的行為,但是并沒有去真正的改變hello()函數的內部實現。
但是,python一直以"優雅"被人追捧,而上述的代碼顯然不夠優雅。
二、優雅的裝飾器
所以,想讓上述裝飾器變得優雅,可以這樣寫:
''' 遇到問題沒人解答?小編創建了一個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),@符號稱為語法糖。
那如果還有其他函數也需要加上類似的裝飾,直接在函數的上方加上@my_decorator就可以,大大提高函數的重復利用與可讀性。
''' 遇到問題沒人解答?小編創建了一個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]三、帶參數的裝飾器
上面的只是一個非常簡單的裝飾器,但是實際場景中,很多函數都是要帶有參數的,比如hello(people_name)。
其實也很簡單,要什么我們就給什么唄,直接在對應裝飾器的wrapper()上,加上對應的參數:
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]但是還沒完,這樣雖然簡單,但是隨之而來另一個問題:因為并不是所有函數參數都是一樣的,
當其他要使用裝飾器的函數參數不止這個一個腫么辦?比如:
沒關系,在python里,*args和**kwargs表示接受任意數量和類型的參數,所以我們可以這樣
寫裝飾器里的wrapper()函數:
同時運行下hello(“老王”),和hello3(“張三”, “李四”),看結果:
這是裝飾后具有的新輸出 你好,老王 ------------------------ 這是裝飾后具有的新輸出 張三對李四說你好! [Finished in 0.1s]上面2種,裝飾器都是接收外來的參數,其實裝飾器還可以接收自己的參數。
比如,我加個參數來控制下裝飾器中打印信息的次數:
注意,這里count裝飾函數中的2個return.
運行下,應該會出現3次:
現在多做一步探索,我們來打印下下面例子中的hello()函數的元信息:
''' 遇到問題沒人解答?小編創建了一個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函數的元信息輸出:
wrapper這說明了,它不再是以前的那個 hello() 函數,而是被 wrapper() 函數取代了。
如果我們需要用到元函數信息,那怎么保留它呢?這時候可以用內置裝飾器@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]四、類裝飾器
裝飾器除了是函數之外,也可以是類。
但是類作為裝飾器的話,需要依賴一個函數__call__(),當調用這個類的實例時,函數__call__()就
會被執行。
來改造下之前的例子,把函數裝飾器改成類裝飾器:
''' 遇到問題沒人解答?小編創建了一個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]跟函數裝飾器一樣,實現一樣的功能。
五、裝飾器的嵌套
既然裝飾器可以增強函數的功能,那如果有多個裝飾器,我都想要怎么辦?
其實,只要把需要用的裝飾器都加上去就好了:
但是要注意這里的執行順序,會從上到下去執行,可以來看下:
''' 遇到問題沒人解答?小編創建了一個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装饰器的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 在Python中用turtle函数画同心
- 下一篇: python 中set集合类型(去重、成