怎么用python寫一個記錄函數執行時間的裝飾器

本篇內容介紹了“怎么用python寫一個記錄函數執行時間的裝飾器”的有關知識，在實際案例的操作過程中，不少人都會遇到這樣的困境，接下來就讓小編帶領大家學習一下如何處理這些情況吧！希望大家仔細閱讀，能夠學有所成！

方法一：用函數實現裝飾器。

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__}執行時間: {time() - start}秒')
        return result
        
    return wrapper

方法二：用類實現裝飾器。類有__call__魔術方法，該類對象就是可調用對象，可以當做裝飾器來使用。

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__}執行時間: {time() - start}秒')
            return result
        
        return wrapper

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

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

from functools import wraps
from time import time


def record_time(output):
    """可以參數化的裝飾器"""
    
    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寫一個記錄函數執行時間的裝飾器”的內容就介紹到這里了，感謝大家的閱讀。如果想了解更多行業相關的知識可以關注億速云網站，小編將為大家輸出更多高質量的實用文章！