Python如何创建装饰器时保留函数元信息
问题
你写了一个装饰器作用在某个函数上，但是这个函数的重要的元信息比如名字、文档字符串、注解和参数签名都丢失了。
解决方案
任何时候你定义装饰器的时候，都应该使用functools库中的@wraps装饰器来注解底层包装函数。例如：
importtime fromfunctoolsimportwraps deftimethis(func): ''' Decoratorthatreportstheexecutiontime. ''' @wraps(func) defwrapper(*args,**kwargs): start=time.time() result=func(*args,**kwargs) end=time.time() print(func.__name__,end-start) returnresult returnwrapper
下面我们使用这个被包装后的函数并检查它的元信息：
>>>@timethis
...defcountdown(n):
...'''
...Countsdown
...'''
...whilen>0:
...n-=1
...
>>>countdown(100000)
countdown0.008917808532714844
>>>countdown.__name__
'countdown'
>>>countdown.__doc__
'\n\tCountsdown\n\t'
>>>countdown.__annotations__
{'n':}
>>>
讨论
在编写装饰器的时候复制元信息是一个非常重要的部分。如果你忘记了使用@wraps，那么你会发现被装饰函数丢失了所有有用的信息。比如如果忽略@wraps后的效果是下面这样的：
>>>countdown.__name__
'wrapper'
>>>countdown.__doc__
>>>countdown.__annotations__
{}
>>>
@wraps有一个重要特征是它能让你通过属性__wrapped__直接访问被包装函数。例如:
>>>countdown.__wrapped__(100000) >>>
__wrapped__属性还能让被装饰函数正确暴露底层的参数签名信息。例如：
>>>frominspectimportsignature >>>print(signature(countdown)) (n:int) >>>
一个很普遍的问题是怎样让装饰器去直接复制原始函数的参数签名信息，如果想自己手动实现的话需要做大量的工作，最好就简单的使用@wraps装饰器。通过底层的__wrapped__属性访问到函数签名信息。
以上就是Python如何创建装饰器时保留函数元信息的详细内容，更多关于Python保留函数元信息的资料请关注毛票票其它相关文章！