2018-04-18-functools模块

mhh4399

于 2019-03-24 00:04:44 发布

阅读量117

点赞数

CC 4.0 BY-SA版权

分类专栏： python进阶

本文链接：https://blog.youkuaiyun.com/github_38970218/article/details/88771406

python进阶专栏收录该内容

9 篇文章

订阅专栏

本文介绍了Python中使用装饰器时遇到的函数签名问题及解决方案。通过使用functools模块中的wraps装饰器，可以轻松地将被装饰函数的元信息如__name__等属性复制到装饰器内部的包装函数中，确保了即使经过装饰器修饰，函数的原始信息依然得以保留。

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

functools模块

@functools.wraps的作用:

把原始函数的__name__等属性复制到wrapper()函数中

>>> def log(func):
...     def wrapper(*args,**kwargs):
...             print("begin Call")
...             func(*args,**kwargs)
...             print("end Call")
...     return wrapper
... 
>>> @log
... def func1():
...     print("running func1")
... 
>>> func1.__name__
'wrapper'
>>> func1()
begin Call
running func1
end Call

因为返回的那个wrapper()函数名字就是'wrapper'，所以，需要把原始函数的__name__等属性复制到wrapper()函数中，否则，有些依赖函数签名的代码执行就会出错。

不需要编写wrapper.__name__ = func.__name__这样的代码，Python内置的functools.wraps就是干这个事的，所以，一个完整的decorator的写法如下：

>>> import functools
>>> def log(func):
...     @functools.wraps(func)
...     def wrapper(*args,**kwargs):
...             print("begin Call")
...             ret = func(*args,**kwargs)
...             print("end Call")
...             return ret
...     return wrapper
... 
>>> @log
... def func2():
...     print("running func2")
... 
>>> func2.__name__
'func2'
>>> func2()
begin Call
running func2
end Call
>>>