您好,欢迎来到爱够旅游网。
搜索
您的当前位置:首页python为被装饰的函数保留元数据方法

python为被装饰的函数保留元数据方法

来源:爱够旅游网

本文主要为大家详细介绍了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)


Copyright © 2019- igbc.cn 版权所有 湘ICP备2023023988号-5

违法及侵权请联系:TEL:199 1889 7713 E-MAIL:2724546146@qq.com

本站由北京市万商天勤律师事务所王兴未律师提供法律服务