python计算时间装饰器_python计算程序执行时间的装饰器使用

1、经常被问程序执行了多久，每个函数都去写很麻烦，所以打算写个装饰器自动计算程序执行时间；推荐文档，大多数基础题都有： https://python3-cookbook.readthedocs.io/zh_CN/latest/

import time

def clock(func):

def clocked(*args, **kwargs):

start = time.time()

result = func(*args, **kwargs)

end = time.time()

print(func.__name__, end - start)

return result

return clocked

@clock

def test():

print("excute...")

time.sleep(5)

test()

执行结果：

"D:\Program Files\python3.6.7\python.exe" D:/pythonWorkspace/untitled4/22.py

excute...

test 5.000607967376709

Process finished with exit code 0

2、装饰器错误使用方法

import datetime

import time

def clock(func):

start = datetime.datetime.now()

print(start)

def clocked(*args, **kwargs):

func(*args, **kwargs)

end = datetime.datetime.now()

total = (end - start).total_seconds()

print(end)

print("执行%s耗时：%d" %(func.__name__, total))

return clocked

@clock

def test():

print("excute...")

time.sleep(5)

test()

错误的执行结果：

"D:\Program Files\python3.6.7\python.exe" D:/pythonWorkspace/untitled4/55.py

2019-08-19 10:24:29.904106

2019-08-19 10:24:29.904106

执行test耗时：0

excute...

Process finished with exit code 0

-------------------------------------------------------------------------------------

适用于例如：

- 验证用户类型

def user_auth(user_group):

def wrapper(func):

def inner(*args, **kwargs):

if user_group == 'SVIP':

print('Dear SVIP')

res = func(*args, **kwargs)

return res

elif user_group == 'General':

res = func(*args, **kwargs)

return res

else:

print('Please login first!')

login()

return inner

return wrapper

@user_auth(user_group='SVIP')

def welcome():

print('Welcome to the index')