python装饰器中的@wraps

2023-01-31 01:01:14 python 装饰 器中

python装饰器中的@wraps

第一段代码:没有添加@wraps

def user_login_confirm(name):
    print('我是', name)
    def deco_fun(view_func):
        print('准备变身。。。')
        def wrapper():
            view_func()
            print('变身完成。。。')
        return wrapper
    return deco_fun

@user_login_confirm('小辣椒')
def tony():
    print('我是', tony.__name__)

tony()

运行结果:

我是 小辣椒
准备变身。。。
我是 wrapper
变身完成。。。

第二段代码:多加一个tony函数

def user_login_confirm(name):
    print('我是', name)
    def deco_fun(view_func):
        print('准备变身。。。')
        def wrapper():
            view_func()
            print('变身完成。。。')
        return wrapper
    return deco_fun

@user_login_confirm('小辣椒')
def tony():
    print('我是', tony.__name__)

def tony():
    print('我是', tony.__name__)

tony()

运行结果:

我是 小辣椒
准备变身。。。
我是 tony

第三段代码:添加了@wraps

from functools import wraps

def user_login_confirm(name):
    print('我是', name)
    def deco_fun(view_func):
        print('准备变身。。。')
        @wraps(view_func)
        def wrapper():
            view_func()
            print('变身完成。。。')
        return wrapper
    return deco_fun

@user_login_confirm('小辣椒')
def tony():
    print('我是', tony.__name__)

tony()

结论:
装饰器将被装饰函数变为了wrapper函数,连函数名也变了,通过@wraps(func)可以让函数名变回来。

相关文章