標準庫functools.wraps的使用方法

首先請看以下代碼

　　# 定義一個最簡單的裝飾器

　　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