Python装饰器应用---统计函数运行时间

from functools import wraps
import time

# 装饰器函数
def fun_run_time(func):
    @wraps(func)
    def inner(*args, **kwargs):
        s_time = time.time()
        ret = func(*args, **kwargs)
        e_time = time.time()
        print("{} cost {} s".format(func.__name__, e_time-s_time))
        return ret
    return inner

# 例子
@fun_run_time
def test():
    for i in range(5):
        time.sleep(2)

# 输出
# test cost 10.001537322998047 s

你可能感兴趣的:(骚操作,Python)