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

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

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

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

先定義一個(gè)簡(jiǎn)單的函數(shù)foo

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

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

import time

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

思考一下，這樣做能夠?qū)崿F(xiàn)，可是如果我還想知道另一個(gè)foo_1的運(yùn)行時(shí)間，那我只能把上面新增加的代碼復(fù)制到foo_1這個(gè)函數(shù)里面,十分麻煩，能不能夠不要改變?cè)镜暮瘮?shù)，即計(jì)算時(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)

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

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

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

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

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

這樣便無需改變?cè)瘮?shù)及其調(diào)用，想計(jì)算函數(shù)foo的用時(shí)，在函數(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ù)時(shí)加上這一行與另外寫foo = cal_time(func):完全等價(jià)
def foo():
    print "call foo()"
 
foo()

如有錯(cuò)漏，請(qǐng)指正

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