Python自学成才之路 装饰器必用的wraps注解

2020-08-20 19:59:28 浏览数 (1)

保留元信息

先看一个案例:

代码语言:javascript复制
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注解。

代码语言:javascript复制
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函数的元信息。

建议写装饰器时都加上wraps注解,这是一个好习惯,反正是没有坏处的。

去除装饰器

Wraps注解还有一个作用就是去除装饰器。上面案例中的greet函数可以绕开装饰器直接执行:

代码语言:javascript复制
greet.__wrapped__()
输出:
say hi

这里装饰器编程部分算是完结了,总结一点:无论是类作为装饰器还是函数作为装饰器,当将被装饰的函数作为参数时,需要返回一个可调用对象,这个可调用对象才是后面真正被调用的。同时,记得加上@wraps注解保证被装饰器修饰的函数元信息保持不变。 题外话: 一开始是想写得不一样一点,看了网上大多数都说wraps注解的作用可以使被装饰的函数保留元信息,确实是这样的,但是为什么要保留元信息呢,我找了个遍也不知道这些元信息保留下来到底有什么作用,或者说找不到一个案例来告诉我不加wraps注解后果会很严重。可惜的是自己水平不够,也不想花太多时间来找这个案例,所以这篇文章算是对装饰器连载的补充吧。后面要是找到了这样的案例,再加到进来!

0 人点赞