關(guān)于python的裝飾器decorator

tinna 發(fā)布于2019-07-30 15:07 / 2097人閱讀

摘要：剛接觸，看到裝飾器相關(guān)的內(nèi)容花了不少時間理解，參考了一些博客，這里做一些總結(jié)。用提供的語法糖來降低字符輸入量。定義函數(shù)時加上這一行與另外寫完全等價如有錯漏，請指正參考資料

剛接觸python，看到裝飾器相關(guān)的內(nèi)容花了不少時間理解，參考了一些博客，這里做一些總結(jié)。

先定義一個簡單的函數(shù)foo

def foo():
    print "call foo()"

現(xiàn)在我們想要計算次函數(shù)執(zhí)行的時間，于是修改代碼如下

import time

def foo():
    start = time.clock()
    print "call foo()"
    end = time.clock()
    print "using time:",end - start

思考一下，這樣做能夠?qū)崿F(xiàn)，可是如果我還想知道另一個foo_1的運(yùn)行時間，那我只能把上面新增加的代碼復(fù)制到foo_1這個函數(shù)里面,十分麻煩，能不能夠不要改變原本的函數(shù)，即計算時間的代碼不加在本來的函數(shù)里面。

這應(yīng)該不難，很容易想到可以這樣寫

import time
 
def foo():
    print "call foo()"
 
def cal_time(func):
    start = time.clock()
    func()
    end =time.clock()
    print "using time:", end - start
 
cal_time(foo)

這樣，如果還需要計算函數(shù)foo_1的時間，只需要cal_time(foo_1)就好了。

但是，這樣其實改變了函數(shù)原本的調(diào)用方式，開始我們是這樣調(diào)用foo()
現(xiàn)在是cal_time(foo)

那么如果現(xiàn)在需要修改foo為foo_plus,我就必須更改n處調(diào)用了foo的代碼！

還有更好的實現(xiàn)方式嗎？能否使調(diào)用foo()產(chǎn)生時調(diào)用cal_time(foo)的效果？
可以想到，把cal_time(foo)的返回值賦給foo，看看下面的實現(xiàn)

import time
 
def foo():
    print "call foo()"
 
# 定義一個計時器，傳入一個函數(shù)，并返回另一個附加了計時功能的方法
def cal_time(func):
     
    # 定義一個內(nèi)嵌的包裝函數(shù)，給傳入的函數(shù)加上計時功能的包裝
    def wrapper():
        start = time.clock()
        func()
        end =time.clock()
        print "using time:", end - start
     
    # 將包裝后的函數(shù)返回
    return wrapper
 
foo = cal_time(foo)
foo()

這樣便無需改變原函數(shù)及其調(diào)用，想計算函數(shù)foo的用時，在函數(shù)調(diào)用foo()前，加上foo = cal_time(foo)即可。

用Python提供的語法糖來降低字符輸入量。

import time
 
def cal_time(func):
    def wrapper():
        start = time.clock()
        func()
        end =time.clock()
        print "using time:", end - start
    return wrapper
 
@cal_time               #定義函數(shù)時加上這一行與另外寫foo = cal_time(func):完全等價
def foo():
    print "call foo()"
 
foo()

如有錯漏，請指正

參考資料：
1.http://www.cnblogs.com/rollen...
2.https://www.liaoxuefeng.com/w...