本文實例講述了Python裝飾器原理與簡單用法。分享給大家供大家參考，具體如下：

今天整理裝飾器，內(nèi)嵌的裝飾器、讓裝飾器帶參數(shù)等多種形式，非常復(fù)雜，讓人頭疼不已。但是突然間發(fā)現(xiàn)了裝飾器的奧秘，原來如此簡單。。。。

第一步 ：從最簡單的例子開始

# -*- coding:gbk -*-
'''示例1: 使用語法糖@來裝飾函數(shù)，相當(dāng)于"myfunc = deco(myfunc)"
但發(fā)現(xiàn)新函數(shù)只在第一次被調(diào)用，且原函數(shù)多調(diào)用了一次'''
def deco(func):
  print("before myfunc() called.")
  func()
  print(" after myfunc() called.")
  return func
@deco
def myfunc():
  print(" myfunc() called.")
myfunc()
myfunc()

這是一個最簡單的裝飾器的例子，但是這里有一個問題，就是當(dāng)我們兩次調(diào)用myfunc()的時候，發(fā)現(xiàn)裝飾器函數(shù)只被調(diào)用了一次。為什么會這樣呢？要解釋這個就要給出破解裝飾器的關(guān)鍵鑰匙了。

這里@deco這一句，和myfunc = deco(myfunc)其實是完全等價的，只不過是換了一種寫法而已

一定要記住上面這句?。。?！

好了，從現(xiàn)在開始，只需要做替換操作就可以了。

將@deco 替換為 myfunc = deco(myfunc)

程序首先調(diào)用deco(myfunc)，得到的返回結(jié)果賦值給了myfunc （注意：在Python中函數(shù)名只是個指向函數(shù)首地址的函數(shù)指針而已）

而deco(myfunc)的返回值就是函數(shù)myfunc()的地址

這樣其實myfunc 沒有變化，也就是說，最后的兩次myfunc()函數(shù)調(diào)用，其實都沒有執(zhí)行到deco()。

有同學(xué)就問了，明明打印了deco()函數(shù)里面的內(nèi)容啊，怎么說沒有調(diào)用到呢。這位同學(xué)一看就是沒有注意聽講，那一次打印是在@deco 這一句被執(zhí)行的。大家親自動手試一下就會發(fā)現(xiàn)" myfunc() called." 這句打印輸出了三次。多的那次就是@deco這里輸出的，因為@deco 等價于myfunc = deco(myfunc)，這里已經(jīng)調(diào)用了deco()函數(shù)了。

第二步 ：確保裝飾器被調(diào)用

怎么解決裝飾器沒有被調(diào)用的問題呢

# -*- coding:gbk -*-
'''示例2: 使用內(nèi)嵌包裝函數(shù)來確保每次新函數(shù)都被調(diào)用，
內(nèi)嵌包裝函數(shù)的形參和返回值與原函數(shù)相同，裝飾函數(shù)返回內(nèi)嵌包裝函數(shù)對象'''
def deco(func):
  def _deco():
    print("before myfunc() called.")
    func()
    print(" after myfunc() called.")
    # 不需要返回func，實際上應(yīng)返回原函數(shù)的返回值
  return _deco
@deco
def myfunc():
  print(" myfunc() called.")
  return 'ok'
myfunc()
myfunc()

這里其實不需要我解釋了，還是按照第一步中的方法做替換就可以了。還是啰嗦幾句吧。。

@deco 替換為 myfunc = deco(myfunc)

程序首先調(diào)用deco(myfunc)，得到的返回結(jié)果賦值給了myfunc ，這樣myfunc 就變成了指向函數(shù)_deco()的指針

以后的myfunc()，其實是調(diào)用_deco()

第三步 ：對帶參數(shù)的函數(shù)進(jìn)行裝飾

破案過程和第一步、第二步完全一致，不再重復(fù)了

# -*- coding:gbk -*-
'''示例5: 對帶參數(shù)的函數(shù)進(jìn)行裝飾，
內(nèi)嵌包裝函數(shù)的形參和返回值與原函數(shù)相同，裝飾函數(shù)返回內(nèi)嵌包裝函數(shù)對象'''
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)

第四步 ：讓裝飾器帶參數(shù)

# -*- coding:gbk -*-
'''示例7: 在示例4的基礎(chǔ)上，讓裝飾器帶參數(shù)，
和上一示例相比在外層多了一層包裝。
裝飾函數(shù)名實際上應(yīng)更有意義些'''
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()

這種帶參數(shù)的裝飾器怎么解釋呢。其實是一樣的，還是我們的替換操作

@deco("mymodule")替換為myfunc = deco("mymodule")(myfunc )

注意啊，這里deco后面跟了兩個括號。

有同學(xué)要問了，這是什么意思？

其實很簡單，先執(zhí)行deco("mymodule")，返回結(jié)果為_deco

再執(zhí)行_deco(myfunc)，得到的返回結(jié)果為__deco

所以myfunc = __deco

破案！

更多關(guān)于Python相關(guān)內(nèi)容可查看本站專題：《Python數(shù)據(jù)結(jié)構(gòu)與算法教程》、《Python Socket編程技巧總結(jié)》、《Python函數(shù)使用技巧總結(jié)》、《Python字符串操作技巧匯總》及《Python入門與進(jìn)階經(jīng)典教程》

希望本文所述對大家Python程序設(shè)計有所幫助。

最新評論