本文實例講述了python 自定義裝飾器。分享給大家供大家參考，具體如下：

先看一個例子

def deco(func):
  print("before myfunc() called.")
  func()
  print("after myfunc() called.")
  return func
@deco
def myfunc():
  print("myfunc() called.")
# myfunc = deco(myfunc) # 與上面的@deco等價
myfunc()
print("***********")
myfunc()

會發(fā)現(xiàn)，輸出為

before myfunc() called.
myfunc() called.
after myfunc() called.
myfunc() called.
***********
myfunc() called.

這就是說，裝飾器里面的東西只調(diào)用了一次，為什么呢？

是因為，在myfunc()函數(shù)的定義前面加一句@deco，本質(zhì)上完全等價于在出現(xiàn)def myfunc()后，先將下面所有內(nèi)容的首地址傳遞給func，然后緊接著加上一句 myfunc = deco(myfunc)。執(zhí)行這句話，表示func代表了本來定義的myfunc()的函數(shù)體，同時函數(shù)myfunc()的地址傳遞給deco()函數(shù)，即 myfunc -> func，這里就相當于myfunc的值與func的值完全相同了。然后執(zhí)行裝飾器里面的內(nèi)容，最后返回給func，傳遞給myfunc。接下來在調(diào)用myfunc()的時候，打印輸出“myfunc() called”。第二次調(diào)用myfunc()函數(shù)的時候，依然只打印輸出“myfunc() called”。為什么第二次沒有執(zhí)行裝飾器里面的內(nèi)容呢？是因為，myfunc = deco(myfunc)這句話只執(zhí)行了一次，而這句話，才是真正執(zhí)行裝飾器里面的內(nèi)容的話。

上面的代碼表示，裝飾器相當于只對第一次調(diào)用他的函數(shù)進行了裝飾，那么，怎么對每次調(diào)用的函數(shù)都裝飾呢？接著看

def deco(func):
  def wrapper(*args, **kwargs): # *args, **kwargs用于接收func的參數(shù)
    print("before myfunc() called.")
    func(*args, **kwargs)
    print("after myfunc() called.")
  return wrapper
@deco
def myfunc(a, b):
  print(a+b)
# myfunc = deco(myfunc) # 與上面的@deco等價
myfunc(1, 2)
print("***********")
myfunc(3, 4)

該代碼輸出結(jié)果為

before myfunc() called.
3
after myfunc() called.
***********
before myfunc() called.
7
after myfunc() called.

我們說了，在myfunc()函數(shù)的定義前面加一句@deco，本質(zhì)上完全等價于在出現(xiàn)def?myfunc()后，先將下面所有內(nèi)容的首地址傳遞給func，然后緊接著加上一句 myfunc = deco(myfunc)。執(zhí)行myfunc(1, 2)命令的時候，myfunc函數(shù)體的地址早已經(jīng)傳遞給了deco()函數(shù)，返回的是wrapper。這是myfunc所代表的地址不再是原來的myfunc的地址，而是wrapper函數(shù)的地址。所以，以后凡是出現(xiàn)myfunc()的地方，都是在調(diào)用wrapper()函數(shù)。即myfunc(1, 2)就是wrapper(1, 2)，所以每次調(diào)用myfunc()時候，裝飾器里面的內(nèi)容都會被執(zhí)行了。而wrapper()函數(shù)體里面的func，就代表了原來myfunc()的函數(shù)體。

怎么進一步理解“在出現(xiàn)def?myfunc()后，先將下面所有內(nèi)容的首地址傳遞給func”這句話呢？看：

def deco(func):
  def wrapper(*args, **kwargs): # *args, **kwargs用于接收func的參數(shù)
    print("wrapper的地址：", wrapper)
    func(*args, **kwargs)
    print("func的地址：", func)
  return wrapper
@deco
def myfunc(a, b):
  print("myfunc的地址：",myfunc)
  print(a+b)
# myfunc = deco(myfunc) # 與上面的@deco等價
myfunc(1, 2)
print("***********")
print("修改后myfunc的地址：",myfunc)

運行結(jié)果：

wrapper的地址： <function deco.<locals>.wrapper at 0x0000023AA9FF58C8>
myfunc的地址： <function deco.<locals>.wrapper at 0x0000023AA9FF58C8>
3
func的地址： <function myfunc at 0x0000023AA9FF5840>
***********
修改后myfunc的地址： <function deco.<locals>.wrapper at 0x0000023AA9FF58C8>

程序執(zhí)行到myfunc(1,2)的時候，本質(zhì)上是在執(zhí)行wrapper(1, 2)，于是先輸出wrapper的地址，再執(zhí)行func()函數(shù)。執(zhí)行func()函數(shù)的時候，輸出myfunc()的地址，（可見，此時myfunc的值與wrapper的是相等），再打印3。當輸出func()函數(shù)的地址，可見func()函數(shù)的地址與myfunc()函數(shù)的地址不一樣了?。。?！這就是說，原來定義的myfunc()函數(shù)的函數(shù)體，已經(jīng)屬于func了，而不屬于myfunc了?。?/p>

進一步見證奇跡??！

def deco(func):
  def wrapper(*args, **kwargs): # *args, **kwargs用于接收func的參數(shù)
    pass
  return wrapper
@deco
def myfunc(a, b):
  print(a+b)
myfunc(1, 2)

該代碼沒有任何輸出。那是因為，執(zhí)行myfunc(1, 2)的時候，本質(zhì)上是執(zhí)行wrapper(1, 2)。而wrapper(1, 2)又不干任何事情，所以沒有輸出。至于print(a+b)這句話，他的地址已經(jīng)屬于func了。

帶參數(shù)的裝飾器，可以參見其他文章

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

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

最新評論