Python-标准库之functools

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返回: {}