Python：funtools模块、覆盖被装饰的函数模块名、名称、限定名、文档、参数注解-优快云博客

本文链接：https://blog.youkuaiyun.com/Smart_look/article/details/115585532

本文详细介绍了Python的functools模块中的update_wrapper和wraps装饰器。update_wrapper用于更新包装函数的特定属性，如模块名、名称、限定名、文档和参数注解。wraps则是一个便捷装饰器，同样用于更新这些属性，并保留原始函数信息。这两个工具在创建自定义装饰器时非常有用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

update_wrapper

def update_wrapper(wrapper, wrapped,assigned = WRAPPER_ASSIGNMENTS,updated = WRAPPER_UPDATES):

类似copy_properties功能
wrapper 包装函数、被更新者，wrapped 被包装函数、数据源
元组WRAPPER_ASSIGNMENTS中是要被覆盖的属性【'__module__', '__name__', '__qualname__', '__doc__', '__annotations__'】模块名、名称、限定名、文档、参数注解
元组WRAPPER_UPDATES中是要被更新的属性，__dict__属性字典
增加一个__wrapped__属性，保留着wrapped函数
import functools

def logger(fn):

    def loger(*args,**kwargs):
        "This funation logger"
        print("函数开始运行")
        ret = fn(*args,**kwargs)
        print("函数结束")
        return ret
    functools.update_wrapper(loger,fn)
    return loger

@logger
def add(x,y):
    '''
    This is funation add
    '''
    return x + y

print(add(3,6),add.__name__,add.__doc__)
print(add.__wrapped__(4,6))   


结果：
函数开始运行
函数结束
9 add 
    This is funation add
    
10

wraps

@functools.wraps(wrapped, assigned=WRAPPER_ASSIGNMENTS, updated=WRAPPER_UPDATES)

类似copy_doc功能
wrapped 被包装函数
元组WRAPPER_ASSIGNMENTS中是要被覆盖的属性【'__module__', '__name__', '__qualname__', '__doc__', '__annotations__'】模块名、名称、限定名、文档、参数注解
元组WRAPPER_UPDATES中是要被更新的属性，__dict__属性字典
增加一个__wrapped__属性，保留着wrapped函
import functools

def logger(fn):
    @functools.wraps(fn)
    def loger(*args,**kwargs):
        "This funation logger"
        print("函数开始运行")
        ret = fn(*args,**kwargs)
        print("函数结束")
        return ret
    return loger

@logger
def add(x,y):
    '''
    This is funation add
    '''
    return x + y

print(add(3,6),add.__name__,add.__doc__)
print(add.__wrapped__(4,6))


结果：
函数开始运行
函数结束
9 add 
    This is funation add
    
10