python:裝飾器

標籤：屬性   執行   nbsp   返回   pre   func   一個   裝飾器   存在   

裝飾器

 

由於函數也是一個對象，而且函數對象可以被賦值給變數，所以，通過變數也能調用該函數。

>>> def now():...     print(‘2015-3-25‘)...>>> f = now>>> f()2015-3-25

函數對象有一個__name__屬性，可以拿到函數的名字：

>>> now.__name__‘now‘>>> f.__name__‘now‘

現在，假設我們要增強now()函數的功能，比如，在函數調用前後自動列印日誌，但又不希望修改now()函數的定義，這種在代碼運行期間動態增加功能的方式，稱之為“裝飾器”（Decorator）。

本質上，decorator就是一個返回函數的高階函數。所以，我們要定義一個能列印日誌的decorator，可以定義如下：

def log(func):    def wrapper(*args, **kw):        print(‘call %s():‘ % func.__name__)        return func(*args, **kw)    return wrapper

觀察上面的log，因為它是一個decorator，所以接受一個函數作為參數，並返回一個函數。我們要藉助Python的@文法，把decorator置於函數的定義處：

@logdef now():    print(‘2015-3-25‘)

調用now()函數，不僅會運行now()函數本身，還會在運行now()函數前列印一行日誌：

>>> now()call now():2015-3-25

把@log放到now()函數的定義處，相當於執行了語句：

now = log(now)

由於log()是一個decorator，返回一個函數，所以，原來的now()函數仍然存在，只是現在同名的now變數指向了新的函數，於是調用now()將執行新函數，即在log()函數中返回的wrapper()函數。

wrapper()函數的參數定義是(*args, **kw)，因此，wrapper()函數可以接受任意參數的調用。在wrapper()函數內，首先列印日誌，再緊接著調用原始函數。

如果decorator本身需要傳入參數，那就需要編寫一個返回decorator的高階函數，寫出來會更複雜。比如，要自訂log的文本：

 

python:裝飾器