ICode9

精准搜索请尝试: 精确搜索
首页 > 编程语言> 文章详细

python中有趣的装饰器

2021-10-07 18:31:07  阅读:186  来源: 互联网

标签:wraps return python testtime func time 有趣 装饰


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

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

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

 

标签:wraps,return,python,testtime,func,time,有趣,装饰
来源: https://www.cnblogs.com/colinshi/p/15375194.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有