【Python基础】装饰器之代码运行时间

代码运行的时间

import datetime
import sys

from tqdm import tqdm
import time
from icecream import ic

def get_run_time(func):
    def wrapper(*args, **kwargs):
        # 定义开始时间
        begin_time = datetime.datetime.now()
        # ret是装饰函数的运行
        ret = func(*args, **kwargs)
        # 定义结束时间
        end_time = datetime.datetime.now()
        # 计算运行时间
        run_time = (end_time - begin_time)
        ic(str(func.__name__) + "函数运行时间为" + str(run_time))
        return ret

    return wrapper

@get_run_time
def test01():
    for i in tqdm(range(100)):
        time.sleep(0.01)

if __name__ == '__main__':
    test01()