原地址:
https://blog.csdn.net/u013858731/article/details/54971762
今天整理裝飾器,內嵌的裝飾器、讓裝飾器帶參數等多種形式,非常複雜,讓人頭疼不已。但是突然間發現了裝飾器的奧祕,原來如此簡單。。。。
第一步 :從最簡單的例子開始
# -*- coding:gbk -*-
'''示例1: 使用語法糖@來裝飾函數,相當於“myfunc = deco(myfunc)”
但發現新函數只在第一次被調用,且原函數多調用了一次'''
def deco(func):
print("before myfunc() called.")
func()
print(" after myfunc() called.")
return func
@deco
def myfunc():
print(" myfunc() called.")
myfunc()
myfunc()
這是一個最簡單的裝飾器的例子,但是這裏有一個問題,就是當我們兩次調用myfunc()的時候,發現裝飾器函數只被調用了一次。爲什麼會這樣呢?要解釋這個就要給出破解裝飾器的關鍵鑰匙了。
這裏@deco這一句,和myfunc = deco(myfunc)其實是完全等價的,只不過是換了一種寫法而已
一定要記住上面這句!!!!
好了,從現在開始,只需要做替換操作就可以了。
將@deco 替換爲 myfunc = deco(myfunc)
程序首先調用deco(myfunc),得到的返回結果賦值給了myfunc (注意:在Python中函數名只是個指向函數首地址的函數指針而已)
而deco(myfunc)的返回值就是函數myfunc()的地址
這樣其實myfunc 沒有變化,也就是說,最後的兩次myfunc()函數調用,其實都沒有執行到deco()。
有同學就問了,明明打印了deco()函數裏面的內容啊,怎麼說沒有調用到呢。這位同學一看就是沒有注意聽講,那一次打印是在@deco 這一句被執行的。大家親自動手試一下就會發現” myfunc() called.” 這句打印輸出了三次。多的那次就是@deco這裏輸出的,因爲@deco 等價於myfunc = deco(myfunc),這裏已經調用了deco()函數了。
第二步 :確保裝飾器被調用
怎麼解決裝飾器沒有被調用的問題呢
# -*- coding:gbk -*-
'''示例2: 使用內嵌包裝函數來確保每次新函數都被調用,
內嵌包裝函數的形參和返回值與原函數相同,裝飾函數返回內嵌包裝函數對象'''
def deco(func):
def _deco():
print("before myfunc() called.")
func()
print(" after myfunc() called.")
# 不需要返回func,實際上應返回原函數的返回值
return _deco
@deco
def myfunc():
print(" myfunc() called.")
return 'ok'
myfunc()
myfunc()
這裏其實不需要我解釋了,還是按照第一步中的方法做替換就可以了。還是囉嗦幾句吧。。
@deco 替換爲 myfunc = deco(myfunc)
程序首先調用deco(myfunc),得到的返回結果賦值給了myfunc ,這樣myfunc 就變成了指向函數_deco()的指針
以後的myfunc(),其實是調用_deco()
第三步 :對帶參數的函數進行裝飾
破案過程和第一步、第二步完全一致,不再重複了
# -*- coding:gbk -*-
'''示例5: 對帶參數的函數進行裝飾,
內嵌包裝函數的形參和返回值與原函數相同,裝飾函數返回內嵌包裝函數對象'''
def deco(func):
def _deco(a, b):
print("before myfunc() called.")
ret = func(a, b)
print(" after myfunc() called. result: %s" % ret)
return ret
return _deco
@deco
def myfunc(a, b):
print(" myfunc(%s,%s) called." % (a, b))
return a + b
myfunc(1, 2)
myfunc(3, 4)
第四步 :讓裝飾器帶參數
# -*- coding:gbk -*-
'''示例7: 在示例4的基礎上,讓裝飾器帶參數,
和上一示例相比在外層多了一層包裝。
裝飾函數名實際上應更有意義些'''
def deco(arg):
def _deco(func):
def __deco():
print("before %s called [%s]." % (func.__name__, arg))
func()
print(" after %s called [%s]." % (func.__name__, arg))
return __deco
return _deco
@deco("mymodule")
def myfunc():
print(" myfunc() called.")
@deco("module2")
def myfunc2():
print(" myfunc2() called.")
myfunc()
myfunc2()
這種帶參數的裝飾器怎麼解釋呢。其實是一樣的,還是我們的替換操作
@deco(“mymodule”)替換爲myfunc = deco(“mymodule”)(myfunc )
注意啊,這裏deco後面跟了兩個括號。
有同學要問了,這是什麼意思?
其實很簡單,先執行deco(“mymodule”),返回結果爲_deco
再執行_deco(myfunc),得到的返回結果爲__deco
所以myfunc = __deco
破案!