直接上代碼看效果:
# 定義一個最簡單的裝飾器
def user_login_data(f):
def wrapper(*args, **kwargs):
return f(*args, **kwargs)
return wrapper
# 用裝飾器裝飾以下兩個函數
@user_login_data
def num1():
print("aaa")
@user_login_data
def num2():
print("bbbb")
if __name__ == '__main__':
print(num1.__name__)
print(num2.__name__)
以上代碼的輸出結果為:
wrapper
wrapper
由此函數使用裝飾器時,函數的函數名即 __name__已經被裝飾器改變.
一般定義裝飾器的話可以不用考慮這點,但是如果多個函數被兩個裝飾器裝飾時就報錯,因為兩個函數名一樣,第二個函數再去裝飾的話就報錯.
解決方案就是引入 functools.wraps ,以上代碼的解決如下:
def user_login_data(f):
@functools.wraps(f)
def wrapper(*args, **kwargs):
return f(*args, **kwargs)
return wrapper
增加@functools.wraps(f), 可以保持當前裝飾器去裝飾的函數的 __name__ 的值不變
以上輸出結果就是:
num1
num2