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)可以讓函數名變回來。