作者：麥?zhǔn)?

來源：麥?zhǔn)寰幊?

裝飾器的本質(zhì)

先簡單說一下什么是裝飾器。

裝飾器，顧名思義，就是起裝飾作用的東西。手機(jī)殼是一種裝飾品。手機(jī)膜也是一種裝飾品。

在編程的世界里，裝飾器的作用，就是用一個(gè)殼子把一個(gè)函數(shù)包起來。以后每次調(diào)用的時(shí)候，實(shí)際上調(diào)用的是這個(gè)殼子。這個(gè)殼子先做一些操作，然后殼子再調(diào)用真正的函數(shù)。

裝飾器可以是一個(gè)函數(shù)，也可以是一個(gè)類。今天就聊一下類裝飾器。

類裝飾器

昨天我們舉了一個(gè)類裝飾器的例子。我把它稍微改造了一下：

• 有一個(gè)函數(shù)cook()，也就是做飯。不是蘋果CEO那個(gè)庫克，是做飯的廚師。
• 我們希望每次做飯之前，都要先洗手。
• 這里的WashHandDecorator就是一個(gè)類裝飾器。
• 我們通過@WashHandDecorator聲明了用這個(gè)裝飾器來裝飾cook()函數(shù)

先看一下代碼和執(zhí)行結(jié)果，思考一下：

class WashHandDecorator(object): def __init__(self, f): self.f = f
      print("洗手裝飾器組裝中...") def __call__(self): print("先洗手，再做飯...")
      self.f() @WashHandDecorator def cook(): print("做飯中...美味香噴噴...但是自己不能吃...")


cook()
cook()
cook()

打印結(jié)果：

洗手裝飾器組裝中...
先洗手，再做飯...
做飯中...美味香噴噴...但是自己不能吃...
先洗手，再做飯...
做飯中...美味香噴噴...但是自己不能吃...
先洗手，再做飯...
做飯中...美味香噴噴...但是自己不能吃...

原理解釋

上面的原理已經(jīng)解釋了一半了，還不夠透徹。所以繼續(xù)：

• WashHandDecorator是裝飾器。
• 它裝飾了函數(shù)cook()以后，以后每次調(diào)用cook()，實(shí)際上是調(diào)用的WashHandDecorator類的實(shí)例，也就是調(diào)用了 __call__函數(shù)，這個(gè)函數(shù)會(huì)再調(diào)用cook()。也就是先洗手，洗完手再去做飯。

其實(shí)裝飾器的語法，也就是那個(gè)@WashHandDecorator，只是一個(gè)語法糖衣。不用這個(gè)語法，我們也可以實(shí)現(xiàn)同樣的效果：

class WashHandDecorator(object): def __init__(self, f): self.f = f
      print("洗手裝飾器組裝中...") def __call__(self): print("先洗手，再做飯...")
      self.f() def cook(): print("做飯中...美味香噴噴...但是自己不能吃...") ## 最關(guān)鍵的一行，魔法就在這里！?。。。?！ cook = WashHandDecorator(cook) 

cook()
cook()
cook()

這個(gè)的執(zhí)行結(jié)果和上面是完全一樣的。

來看最關(guān)鍵的這一行cook = WashHandDecorator(cook)：

• 參數(shù)中的cook代表的是上面定義的cook()函數(shù)，它作為參數(shù)傳給了WashHandDecorator的構(gòu)造函數(shù)。
• 而前面的變量名cook是新構(gòu)造出來的WashHandDecorator的實(shí)例，只是故意讓它的名字和函數(shù)名相同。
• 這樣后面再調(diào)用cook()的時(shí)候，實(shí)際上不是調(diào)用的cook()函數(shù)，而是調(diào)用的這個(gè)對(duì)象實(shí)例。
• 對(duì)象一般不可以直接這樣cook()調(diào)用的，除非它有__call__函數(shù)，而WashHandDecorator正好有這個(gè)函數(shù)。
• __call__函數(shù)里實(shí)現(xiàn)了：先洗手，然后再調(diào)用原本的cook()函數(shù)。
• 這就是整個(gè)裝飾的過程。使用裝飾器的語法不過是自動(dòng)幫我們實(shí)現(xiàn)了上面的過程。
• 所以一個(gè)類要成為裝飾的，最重要的就是必須有__call__函數(shù)，這就和上一個(gè)三分鐘呼應(yīng)上了。

懂了嗎？沒懂再看一遍。如果真懂了，有沒有一種恍然大悟，甚至醍醐灌頂?shù)母杏X？