您的位置：首页 > 编程语言 > Python开发

python如何为被装饰的函数保留元数据

2018-03-21 11:14 288 查看

本文实例为大家分享了python为被装饰的函数保留元数据的具体代码，供大家参考，具体内容如下

案例：

       在函数对象中保存着一些函数的元数据，如：

              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 函数 元数据
相关文章推荐
新的分享
章节导航