python中有趣的装饰器

在函数上添加一个装饰器，增加额外的操作处理(比如日志、计时等)。

特别是在代码调试阶段特别好用。

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 原理也是一样的。