一個例子解釋python裝飾器

談到python裝飾器,一般都先提到函數(shù)式編程,但是其實運用python裝飾器,和知不知道他是函數(shù)式編程沒啥關(guān)系。

例子

def decorator(func):
    def wrapper(*args, **kwargs):
        print('我是裝飾器呦')
        return  func(*args, **kwargs)
    return wrapper
    
@decorator
def test():
    print('我是test')
    
test()

輸出如下:

我是裝飾器呦
我是test

其實就這么簡單,完全可以把代碼里的@理解為我們聊天中用的@,在調(diào)用某一個函數(shù)的時候@你想調(diào)用的裝飾器函數(shù),然后就可以在函數(shù)自身代碼不變的情況下、增添一些具有普適性的功能。

應(yīng)用

那么裝飾器能做什么呢?emmm。。。最簡單的一個應(yīng)用,計算函數(shù)運行時間啊。

import time

def run_time(func):
    def wrapper(*args, **kw):
        t = time.time()
        ans = func(*args, **kw)
        t = time.time() - t
        return ans, t
    return wrapper
    
@run_time
def test():
    x=1
    for i in range(10 ** 6):
        x += 1
    return x
    
test()

輸出如下:

(1000001, 0.08555388450622559)

曾幾何時,我為了計算每種目標跟蹤算法處理一幀圖片的時間,寫了多少重復(fù)代碼,這不是優(yōu)雅多了==!

當然,裝飾器的應(yīng)用不止如此,它還可以傳遞參數(shù),著名的用法比如web框架flask的路由配置,有興趣可以閱讀一下它的源碼,很漂亮。

所需基礎(chǔ)

為了徹底理解裝飾器,有兩個基礎(chǔ)知識要鞏固一下。一個是閉包函數(shù),一個是可變參數(shù)。這里放上兩個鏈接,個人覺得講得比較好。
https://www.liaoxuefeng.com/wiki/0014316089557264a6b348958f449949df42a6d3a2e542c000/001431752945034eb82ac80a3e64b9bb4929b16eeed1eb9000

https://foofish.net/python-closure.html

補充

  • 我們給上面的裝飾器加一個參數(shù):
import time

def run_time(text):
    def decorator(func):
        def wrapper(*args, **kwargs):
            t = time.time()
            ans = func(*args, **kwargs)
            t = time.time() - t
            if t>text:
                return ans, '太慢'
            else:
                return ans, '正常'    
        return wrapper
    return decorator
    
@run_time(1)
def test():
    x=1
    for i in range(10 ** 6):
        x += 1
    return x
    
test( )
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容