Python裝飾器實現函數運行時間的計算

Posted on by

個人理解

裝飾器： 通過閉包和將一個函數作為另一個函數參數的形式，實現已有功能的靈活調用

例如：

首先設置瞭一個time_master的計時器函數，在運行某個函數的同時，用來統計函數的耗時

那麼，如果想知道函數性能, 每次寫完新的函數後，都放到time_master函數中運行一次來統計。

——> 這是比較麻煩的。相當於雖然有瞭一個稱，但是所有的新買食材都需要逐個放到稱上去稱一遍，來獲得食材的重量

——> 如果能夠省略掉一次一次上稱這個步驟就好瞭

——> 比如每個食材進門的同時就從稱上走過，那進來的同時，也就有瞭重量的數據

例子：調用函數的同時對函數進行計時

實現方法1：@語法糖

代碼：

# 涉及到計時，需要引入time模塊
import time
# 定義一個對函數運行耗時統計的計時器，用func作為形參來代替需要統計的函數
def time_master(func):
    def call_func():
        print("計時器開始：函數開始調用：")
        start_time = time.time()
        func()
        end_time = time.time()
        print('計時器結束，函數調用完成')
        return print(f'計時器結果返回：函數調用耗時{end_time-start_time:.2f}')
    return call_func
@time_master 
# 即在調用myfunc函數時，不是直接調用myfunc
# 而是將myfunc函數作為一個參數放入到@的裝飾器中，然後去調用裝飾器
def myfunc():
    time.sleep(2)
    print('myfunc函數運行')
    time.sleep(4)
    print('myfunc函數運行結束')
myfunc()  # 調用myfunc

實現結果：

計時器開始：函數開始調用：
myfunc函數運行
myfunc函數運行結束
計時器結束，函數調用完成
計時器結果返回：函數調用耗時6.01

實現方法2：閉包

可以理解成，本來定義瞭一個myfunc的函數，但這個函數本身沒有計時的功能，而恰巧有一個time_master的函數，在運行子函數的同時，還會對子函數進行計時

因此，通過重新定義 myfunc = time_master(myfunc)， 即將myfunc作為參數傳入到time_master中，作為myfunc函數的新定義

代碼：

import time
def time_master(func):
    def call_func():
        print("計時器開始：函數開始調用：")
        start_time = time.time()
        func()
        end_time = time.time()
        print('計時器結束，函數調用完成')
        print(f'計時器結果返回：函數調用耗時{end_time-start_time:.2f}')
    return call_func
def myfunc():
    time.sleep(2)
    print('myfunc函數運行')
    time.sleep(4)
    print('myfunc函數運行結束')
myfunc = time_master(myfunc)  # 和實現方法一的差距就在於是在myfunc前面@裝飾器，還是在後面對myfunc函數進行二次定義
myfunc()

實現結果：

計時器開始：函數開始調用：
myfunc函數運行
myfunc函數運行結束
計時器結束，函數調用完成
計時器結果返回：函數調用耗時6.01

實現方式1和 2的差異

實現方式1和實現方式2: 其實沒什麼區別，無非是一開始就用@time_master來規定，還是定義完myfunc之後，再多做一步讓myfun放入time_master中去執行

總結

本篇文章就到這裡瞭，希望能夠給你帶來幫助，也希望您能夠多多關註WalkonNet的更多內容!               

推薦閱讀：