怎么用python寫一個(gè)記錄函數(shù)執(zhí)行時(shí)間的裝飾器

本篇內(nèi)容介紹了“怎么用python寫一個(gè)記錄函數(shù)執(zhí)行時(shí)間的裝飾器”的有關(guān)知識，在實(shí)際案例的操作過程中，不少人都會遇到這樣的困境，接下來就讓小編帶領(lǐng)大家學(xué)習(xí)一下如何處理這些情況吧！希望大家仔細(xì)閱讀，能夠?qū)W有所成！

方法一：用函數(shù)實(shí)現(xiàn)裝飾器。

from functools import wraps
from time import time


def record_time(func):
    
    @wraps(func)
    def wrapper(*args, **kwargs):
        start = time()
        result = func(*args, **kwargs)
        print(f'{func.__name__}執(zhí)行時(shí)間: {time() - start}秒')
        return result
        
    return wrapper

方法二：用類實(shí)現(xiàn)裝飾器。類有__call__魔術(shù)方法，該類對象就是可調(diào)用對象，可以當(dāng)做裝飾器來使用。

from functools import wraps
from time import time


class Record:
    
    def __call__(self, func):
        
        @wraps(func)
        def wrapper(*args, **kwargs):
            start = time()
            result = func(*args, **kwargs)
            print(f'{func.__name__}執(zhí)行時(shí)間: {time() - start}秒')
            return result
        
        return wrapper

說明：裝飾器可以用來裝飾類或函數(shù)，為其提供額外的能力，屬于設(shè)計(jì)模式中的代理模式。

擴(kuò)展：裝飾器本身也可以參數(shù)化，例如上面的例子中，如果不希望在終端中顯示函數(shù)的執(zhí)行時(shí)間而是希望由調(diào)用者來決定如何輸出函數(shù)的執(zhí)行時(shí)間，可以通過參數(shù)化裝飾器的方式來做到，代碼如下所示。

from functools import wraps
from time import time


def record_time(output):
    """可以參數(shù)化的裝飾器"""
    
    def decorate(func):
        
        @wraps(func)
        def wrapper(*args, **kwargs):
            start = time()
            result = func(*args, **kwargs)
            output(func.__name__, time() - start)
            return result
            
        return wrapper
    
    return decorate

“怎么用python寫一個(gè)記錄函數(shù)執(zhí)行時(shí)間的裝飾器”的內(nèi)容就介紹到這里了，感謝大家的閱讀。如果想了解更多行業(yè)相關(guān)的知識可以關(guān)注億速云網(wǎng)站，小編將為大家輸出更多高質(zhì)量的實(shí)用文章！