实现:
def timer(func):
def wrapper(*args, **kw):
tic = time()
res = func(*args, **kw)
toc = time()
print('[timer] {} cost {:.2f}s'.format(func.__name__, toc - tic))
return res
return wrapper
[注] 如果 func 函数有返回,wrapper 里也要返回其输出
使用:
@timer
def show(message):
print(message)
show('hello')