裝飾器主要是用來包裝函數,對於一些常用的功能,譬如:日志打印,函數計時,身份認證。我們可以使用裝飾器來實現,這樣可以降低整個程序的復雜度和減少程序的代碼量。
它實際上就是函數,不同的是,它把一個函數當做參數,然後返回一個替代版函數。
下面看一個簡單的示例:
def add_number(func): def adder(arg): return func(arg)+100 return adder
def f(x): return x f=add_number(f) print f(20)
add_number就是一個裝飾器函數,它接受一個函數(f)作為參數,然後返回另外一個函數(adder)賦值給原來的函數,這樣,原來的函數不用新添加額外的代碼量而實現了加法的功能。
這個就是裝飾器的原始實現。
But,這種方式還是有點不太方便,畢竟還是繞了一圈,用f=add_number(f)來給原來的函數重新賦值。
其實,Python中可以用下列方式來簡化對於裝飾器的引用。
def add_number(func): def adder(arg): return func(arg)+100 return adder @add_number def f(x): return x print f(20)
只需一個簡單的@add_numbe調用,是不是方便,簡單了很多,基本上沒侵入原來的代碼。
額,大家發現沒有,作為裝飾器,每次接受的參數無非兩種:函數和函數的參數,但書寫的格式基本一樣,有沒有辦法來簡化這種書寫呢?
有,Python提供了一個decorator包,可以大大簡化裝飾器的書寫。
So,第三種實現方式為:
from decorator import decorator @decorator def wrapper(func,arg): return func(arg)+100 @wrapper def f(x): return x print f(20)
喔,果然更加簡單了~
以上示例接受的都是一個參數,其實,函數本身是可以接受可變參數的。
如:
@decorator def wrapper(f,arg1,*args,**kwargs): print "I am just a wrapper~" return f(arg1,*args,**kwargs)
@wrapper def f(arg1,*args,**kwargs): print arg1 for eacheArg in args: print 'non-keyword arg:',eacheArg for eachKw in kwargs.keys(): print 'keyword arg: %s:%d' % (eachKw,kwargs[eachKw])
args=('Joy','Steve') kwargs={"age":20} f('China',*args,**kwargs)
輸出結果為:
I am just a wrapper~ China non-keyword arg: Joy non-keyword arg: Steve keyword arg: age:20
關於*args,**kwargs的區別,兩者都可用於表示可變長度的參數。只不過前者是用元祖表示,沒有key值,後者是字典,有key值。兩者可用於在同一個函數中,但是,*args必須出現在**kwargs之前。
譬如下例:
def test_var_args_call(arg1, arg2, arg3): print "arg1:", arg1 print "arg2:", arg2 print "arg3:", arg3 args=(1,2,3) kwargs ={"arg1":"1","arg3": 3, "arg2": "2"} test_var_args_call(*args) print '-----------------' test_var_args_call(**kwargs)
兩者的實現效果一樣。
最後來個示例,通過顯示函數執行的時間來裝飾一個函數
import time def log(func): def wrapper(*args, **kw): print '[%s] %s() was called...' % (time.ctime(),func.__name__) return func(*args, **kw) return wrapper @log def foo(): pass for i in range(4): foo() time.sleep(1)
輸出結果如下:
[Wed Jul 27 09:17:23 2016] foo() was called... [Wed Jul 27 09:17:24 2016] foo() was called... [Wed Jul 27 09:17:25 2016] foo() was called... [Wed Jul 27 09:17:26 2016] foo() was called...
Ubuntu 14.04安裝Python 3.3.5 http://www.linuxidc.com/Linux/2014-05/101481.htm
CentOS上源碼安裝Python3.4 http://www.linuxidc.com/Linux/2015-01/111870.htm
《Python核心編程 第二版》.(Wesley J. Chun ).[高清PDF中文版] http://www.linuxidc.com/Linux/2013-06/85425.htm
《Python開發技術詳解》.( 周偉,宗傑).[高清PDF掃描版+隨書視頻+代碼] http://www.linuxidc.com/Linux/2013-11/92693.htm
Python腳本獲取Linux系統信息 http://www.linuxidc.com/Linux/2013-08/88531.htm
在Ubuntu下用Python搭建桌面算法交易研究環境 http://www.linuxidc.com/Linux/2013-11/92534.htm
Python 語言的發展簡史 http://www.linuxidc.com/Linux/2014-09/107206.htm
Python 的詳細介紹:請點這裡
Python 的下載地址:請點這裡