python裝飾器中的@wraps

python裝飾器中的@wraps

第一段代碼：沒(méi)有添加@wraps

def user_login_confirm(name):
    print('我是', name)
    def deco_fun(view_func):
        print('準(zhǔn)備變身。。。')
        def wrapper():
            view_func()
            print('變身完成。。。')
        return wrapper
    return deco_fun

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

tony()

運(yùn)行結(jié)果：

我是 小辣椒
準(zhǔn)備變身。。。
我是 wrapper
變身完成。。。

第二段代碼：多加一個(gè)tony函數(shù)

def user_login_confirm(name):
    print('我是', name)
    def deco_fun(view_func):
        print('準(zhǔn)備變身。。。')
        def wrapper():
            view_func()
            print('變身完成。。。')
        return wrapper
    return deco_fun

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

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

tony()

運(yùn)行結(jié)果：

我是 小辣椒
準(zhǔn)備變身。。。
我是 tony

第三段代碼：添加了@wraps

from functools import wraps

def user_login_confirm(name):
    print('我是', name)
    def deco_fun(view_func):
        print('準(zhǔn)備變身。。。')
        @wraps(view_func)
        def wrapper():
            view_func()
            print('變身完成。。。')
        return wrapper
    return deco_fun

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

tony()

結(jié)論：
裝飾器將被裝飾函數(shù)變?yōu)榱藈rapper函數(shù)，連函數(shù)名也變了，通過(guò)@wraps(func)可以讓函數(shù)名變回來(lái)。