码迷,mamicode.com
首页 > 编程语言 > 详细

python_为被装饰的函数保留元数据

时间:2017-08-05 22:55:08      阅读:277      评论:0      收藏:0      [点我收藏+]

标签:default   **kwargs   时间   name   运行   返回   保存   nts   ==   

案例:

       在函数对象中保存着一些函数的元数据,如:

              f.__name__           函数名

              f.__doc__              函数文档

              f.__moudle__       函数所属模块名

              f.__dict__              属性字典

              f.__defaults__       默认参数组

              ……

       在使用装饰器后,在装饰器里访问以上属性时,我们看到的是装饰器函数的元数据

       需求:

              实现在装饰器函数中,保留 被装饰函数 的元数据

如何实现?

       通过 functools中的wraps, update_wrapper方法实现,每个都可单独实现

#!/usr/bin/python3

import time
from functools import (wraps, update_wrapper, WRAPPER_ASSIGNMENTS, WRAPPER_UPDATES)


def count_time(func):
    """
    给目标函数加上计算运行时间统计
    """
    # 这个装上器和update_wrapper一样,默认参数WRAPPER_ASSIGNMENTS, WRAPPER_UPDATES
    @wraps(func)
    def wrapper(*args, **kwargs):
        start_time = time.time()
        
        # 定义result接收函数返回值,并且在装饰函数最后返回回去
        resutl = func(*args, **kwargs)
        print(‘运行时间:‘, time.time()-start_time)
        return resutl
    
    # 其中默认参数 WRAPPER_ASSIGNMENTS, WRAPPER_UPDATES
    # update_wrapper(wrapper, func)
    return wrapper


@count_time
def add(num=100):
    """
    计算 0~num 累加值,默认num=100
    """
    time.sleep(1)
    return sum([x for x in range(num+1)])

if __name__ == ‘__main__‘:
    print(‘函数名:‘, add.__name__)
    print(‘属性字典:‘, add.__dict__)
    print(‘函数默认参数:‘, add.__defaults__)
    print(‘函数所在模块:‘, add.__module__)
    print(‘函数文档:‘, add.__doc__)
    
    # 打印两个默认参数
    print(WRAPPER_ASSIGNMENTS, WRAPPER_UPDATES)
    result = add()
    print(result)

  

python_为被装饰的函数保留元数据

标签:default   **kwargs   时间   name   运行   返回   保存   nts   ==   

原文地址:http://www.cnblogs.com/2bjiujiu/p/7291623.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!