import time

def run_time(func):
  def wrapper(*args, **kwargs):
    start = time.time()
    r = func(*args, **kwargs)
    print time.time() - start
    return r
  return wrapper

我們用這個裝飾器裝飾一個test函數(shù)

@run_time
def test():
  print "just a test"

前面說過其實裝飾器就是一個語法糖, 就是將被裝飾的函數(shù)作為參數(shù)傳遞給裝飾器函數(shù), 所以上面可以展開為

test = run_time(test)

裝飾器將在解釋器運行一開始就被加載, 從而將被裝飾的函數(shù)將被展開成如上方式, 因為 run_time裝飾器返回wrapper函數(shù), 所以當調(diào)用test函數(shù)時其實就是對wrapper的調(diào)用

如果你在Python shell下執(zhí)行以上語句就會發(fā)現(xiàn)定義完test函數(shù)然后查看test時, shell所展示的是wrapper函數(shù)

201542492457922.png (502×346)

接下來說說如何編寫帶參數(shù)的裝飾器, 大家如果細心的話就可以發(fā)現(xiàn)其實帶參數(shù)的裝飾器是經(jīng)過調(diào)用"裝飾器"函數(shù)返回的一個裝飾器, 之所以裝飾器上打引號是說明其實這個所謂的"裝飾器"只不過是一個普通的函數(shù), 但這個普通的函數(shù)返回一個裝飾器, 可以參看下面例子:

import time

def route(url):
  def decorator(func):
    func.__url__ = url
    return func
  return decorator

@route(r"/")
def index():
  return "Hi"

大家可以發(fā)現(xiàn)在使用route裝飾器時我們其實是調(diào)用了route函數(shù), route函數(shù)返回一個decorator裝飾器, 因為我們不需要在裝飾器內(nèi)運行函數(shù), 所以不需要一個wrapper函數(shù)來收集參數(shù).

以上就是全部內(nèi)容, 希望對裝飾器一知半解的人有些許幫助

您可能感興趣的文章: