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)