歡迎來到Linux教程網
Linux教程網
Linux教程網
Linux教程網
您现在的位置: Linux教程網 >> UnixLinux >  >> Linux編程 >> Linux編程

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

Copyright © Linux教程網 All Rights Reserved