Python裝飾器詳解

裝飾器主要是用來包裝函數，對於一些常用的功能，譬如：日志打印，函數計時，身份認證。我們可以使用裝飾器來實現，這樣可以降低整個程序的復雜度和減少程序的代碼量。

它實際上就是函數，不同的是，它把一個函數當做參數，然後返回一個替代版函數。

下面看一個簡單的示例：

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 的下載地址：請點這裡