保留元信息
先看一个案例:
def my_decorate(func):
def inner_fun(*args, **kargs):
print('here is my_decorate')
func()
return inner_fun
@my_decorate
def greet():
print("say hi")
print(greet.__name__)
输出:
inner_fun
Greet函数的名字为何会变成了inner_fun?因为装饰器最终会返回一个可调用对象,而这个可调用对象才是正在最后被执行的,所以greet被装饰器修饰后,得到的是inner_fun这个函数(函数是可调用对象),所以最终greet就变成了inner_fun的一部分,于是乎greet.__name__实际上是inner_fun.name。那该怎么解决这个问题呢,加上wraps注解。
from functools import wraps
def my_decorate(func):
@wraps(func)
def inner_fun(*args, **kargs):
print('here is my_decorate')
func()
return inner_fun
@my_decorate
def greet():
print("say hi")
print(greet.__name__)
输出:
greet
这里wraps本身也是一个装饰器,这个装饰器是一个带参数的装饰器,参数本身就是greet函数。在获取greet函数元信息时,实际上还是在执行inner_fun.name,但是inner_fun有wraps装饰器,最终元信息是通过wraps装饰器返回的,wraps装饰器对greet函数做了份拷贝,所以拿到的还是greet函数的元信息。