python 计算函数执行时间装饰器(带参数装饰器写法)

import time
import functools
DEFAULT_FMT = '[{elapsed:0.8f}s] {name}({args}) -> {result}'
def clock(fmt=DEFAULT_FMT): # <1>
def decorate(func): # <2>
@functools.wraps(func)
def clocked(*args, *kwargs):
t0 = time.time()
result = func(args, **kwargs)
elapsed = time.time() - t0
name = func.name
arg_lst = []
if args:
arg_lst.append(', '.join(repr(arg) for arg in args))
if kwargs:
pairs = ['%s=%r' % (k, w) for k, w in sorted(kwargs.items())]
arg_lst.append(', '.join(pairs))
arg_str = ', '.join(arg_lst)
print('[%0.8fs] %s(%s) -> %r ' % (elapsed, name, arg_str, result))
return result
return clocked
return decorate

@functools.lru_cache() # ➊
@clock() # ➋
def fibonacci(n):
if n < 2:
return n
return fibonacci(n-2) + fibonacci(n-1)
if name=='main':
print(fibonacci(6))