python装饰器保留函数原有信息的@functools.wraps源码解析

最新推荐文章于 2022-05-14 00:02:37 发布

qq_37440689

最新推荐文章于 2022-05-14 00:02:37 发布

阅读量392

点赞数

本文链接：https://blog.csdn.net/qq_37440689/article/details/89527217

版权

本文深入探讨Python中的装饰器，并详细解析`functools.wraps`的作用，阐述如何利用它来保留被装饰函数的元信息，如名称、文档字符串和模块等关键细节。

摘要由CSDN通过智能技术生成

python装饰器保留函数原有信息的@functools.wraps源码解析

# 调用functools库的partial方法
from functools import partial

# 要保留的原函数参数
WRAPPER_ASSIGNMENTS = ('__module__', '__name__', '__qualname__', '__doc__',
                       '__annotations__')
# 更新属性字典
WRAPPER_UPDATES = ('__dict__',)


def update_wrapper(wrapper,
                   wrapped,
                   assigned=WRAPPER_ASSIGNMENTS,
                   updated=WRAPPER_UPDATES):
    '''
    恢复原函数自身属性的装饰器实际处理函数
    :param wrapper:装饰器的闭包函数
    :param wrapped:被装饰的原函数
    :param assigned:需要保留的原函数属性集合
    :param updated:更新闭包函数的属性
    :return:恢复原函数属性的闭包函数
    '''
    print("wrapper is", wrapper.__name__)
    #循环恢复原函数的属性
    for attr in assigned:
        try:
            value = getattr(wrapped, attr) #获取原函数的属性
        except AttributeError: