4、python函數(shù)是一個對象，函數(shù)執(zhí)行完后內(nèi)部變量被回收；return的東西不會被回收。

由于函數(shù)也是一個對象，而且函數(shù)對象可以被賦值給變量，所以，通過變量也能調(diào)用該函數(shù)。

>>> def now(): ... print('2015-3-25') ... >>> f = now >>> f() 2015-3-25# 函數(shù)對象有一個__name__屬性，可以拿到函數(shù)的名字： >>> now.__name__ 'now' >>> f.__name__ 'now'

現(xiàn)在，假設(shè)我們要增強(qiáng)now()函數(shù)的功能，比如，在函數(shù)調(diào)用前后自動打印日志，但又不希望修改now()函數(shù)的定義，這種在代碼運(yùn)行期間動態(tài)增加功能的方式，稱之為“裝飾器”（Decorator）。

本質(zhì)上，decorator就是一個 返回值為函數(shù)的 高階函數(shù)。所以，我們要定義一個能打印日志的decorator，可以定義如下：

# 注意格式 def log(func):def wrapper(*args, **kw):print('call %s():' % func.__name__)return func(*args, **kw) #返回函數(shù)帶參數(shù)return wrapper #返回函數(shù)名

觀察上面的log，因為它是一個decorator，所以接受一個函數(shù)作為參數(shù)，并返回一個函數(shù)。我們要借助Python的@語法，把decorator置于函數(shù)的定義處：

@log def now():print('2015-3-25')

調(diào)用now()函數(shù)，不僅會運(yùn)行now()函數(shù)本身，還會在運(yùn)行now()函數(shù)前打印一行日志：

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

當(dāng)執(zhí)行@log語句時，其實是進(jìn)行了以下操作:
1.調(diào)用函數(shù) log(now) 并返回wrapper函數(shù)；
2.對now 重新賦值，令now = wapper
當(dāng)調(diào)用now時即 now()，其實是調(diào)用了wapper函數(shù)，而在wapper函數(shù)內(nèi)部又調(diào)用了原來的now函數(shù)。

如果decorator本身需要傳入?yún)?shù)，那就需要編寫一個返回decorator的高階函數(shù)，寫出來會更復(fù)雜。比如，要自定義log的文本：

def log(text):def decorator(func):def wrapper(*args, **kw):print('%s %s():' % (text, func.__name__))return func(*args, **kw)return wrapperreturn decorator#這個3層嵌套的decorator用法如下： @log('execute') def now():print('2015-3-25')# 執(zhí)行結(jié)果如下： >>> now() execute now(): 2015-3-25

和兩層嵌套的decorator相比，3層嵌套的效果是這樣的：

now = log(‘execute’)(now)

我們來剖析上面的語句，首先執(zhí)行l(wèi)og(‘execute’)，返回的是decorator函數(shù)，再調(diào)用返回的函數(shù)，參數(shù)是now函數(shù)，返回值最終是wrapper函數(shù)。

以上兩種decorator的定義都沒有問題，但還差最后一步。因為我們講了函數(shù)也是對象，它有name等屬性，但你去看經(jīng)過decorator裝飾之后的函數(shù)，它們的name已經(jīng)從原來的’now’變成了’wrapper’：

>>> now.__name__ 'wrapper'

因為返回的那個wrapper()函數(shù)名字就是’wrapper’，所以，需要把原始函數(shù)的name等屬性復(fù)制到wrapper()函數(shù)中，否則，有些依賴函數(shù)簽名的代碼執(zhí)行就會出錯。
不需要編寫wrapper.name = func.name這樣的代碼，Python內(nèi)置的functools.wraps就是干這個事的，所以，一個完整的decorator的寫法如下：

import functoolsdef log(func):@functools.wraps(func)def wrapper(*args, **kw):print('call %s():' % func.__name__)return func(*args, **kw)return wrapper# 或者針對帶參數(shù)的decorator： import functoolsdef log(text):def decorator(func):@functools.wraps(func)def wrapper(*args, **kw):print('%s %s():' % (text, func.__name__))return func(*args, **kw)return wrapperreturn decorator

總結(jié)

以上是生活随笔為你收集整理的Pyhton函数式编程简介（四）装饰器的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。