在函数上添加一个装饰器,增加额外的操作处理(比如日志、计时等)。
特别是在代码调试阶段特别好用。
import time from functools import wraps def funcruntime(func): @wraps(func) def wrapper(*args, **kwargs): start = time.time() result = func(*args, **kwargs) end = time.time() print(func.__name__, end-start) return result return wrapper
简单的时间装饰器。
注:这里特别要注意的是需要导入functools里@wraps 装饰器来注解底层包装函数。非常重要,不然原来的函数里的元信息比如名字、文档字符串、注解和参数签名都丢失了。其实任何装饰器操作需要保留元数据都应该使用这个@wraps 装饰器,甚至我觉得这个就应该无脑用。
@funcruntime def testtime(t: int): """ 测试装饰器效果 :param t: int :return: None """ time.sleep(t) return testtime(1)
执行效果如下:
这样一个简单的测试方法的装饰器就完成了。
装饰器其实就是把func直接当做参数传入做了一个闭包处理。
def testtime(t: int): """ 测试装饰器效果 :param t: int :return: None """ time.sleep(t) return testtime = funcruntime(testtime) testtime(1)
这段代码和上面是完全等价的。
注:内置的装饰器 @staticmethod,@classmethod,@property
原理也是一样的。