functools模块用于高阶函数,即参数或返回值为其他函数的函数,此模块的功能适用于所有可调用对象
- @wraps装饰器:任何时候定义装饰器,都应使用wraps,可以保留装饰的函数的元信息:比如名字,文档字符串,注解和参数签名
from functools import wraps
import time
def runtime(func):
@wraps(func)
def wrapper(*args,**kwargs):
start = time.time()
r = func(*args, **kwargs)
end = time.time()
print('{}-{}'.format(func.__name__, end - start))
return r
return wrapper
@runtime
def countdown(n):
'''这是一个测试'''
while n > 0:
n-=1
print(countdown(1000) )
#返回结果:countdown-4.601478576660156e-05
print(countdown.__name__)
#不加wraps返回None wrapper,加了wraps返回:countdown
print(countdown.__doc__)
#如果不加wraps返回None,加了wraps返回:这是一个测试
print(countdown.__annotations__)
#如果不加wraps返回{},加了wraps返回: {}