直接上代碼看效果:

　　# 定義一個(gè)最簡(jiǎn)單的裝飾器

　　def user_login_data(f):
　　　　def wrapper(*args, **kwargs):
　　　　　　return f(*args, **kwargs)

　　　　return wrapper

　　# 用裝飾器裝飾以下兩個(gè)函數(shù)
　　
　　@user_login_data
　　def num1():
　　　　print("aaa")



　　@user_login_data
　　def num2():
　　　　print("bbbb")

　　if __name__ == '__main__':
　　　　print(num1.__name__)
　　　　print(num2.__name__)

　以上代碼的輸出結(jié)果為:

　　　　wrapper

　　　　wrapper

由此函數(shù)使用裝飾器時(shí),函數(shù)的函數(shù)名即 __name__已經(jīng)被裝飾器改變.

一般定義裝飾器的話可以不用考慮這點(diǎn),但是如果多個(gè)函數(shù)被兩個(gè)裝飾器裝飾時(shí)就報(bào)錯(cuò),因?yàn)閮蓚€(gè)函數(shù)名一樣,第二個(gè)函數(shù)再去裝飾的話就報(bào)錯(cuò).

解決方案就是引入  functools.wraps  ,以上代碼的解決如下: 

　　　　

　　　　def user_login_data(f):
　　　　　　@functools.wraps(f)
　　　　　　def wrapper(*args, **kwargs):
　　　　　　　　return f(*args, **kwargs)

　　　　　　return wrapper

增加@functools.wraps(f), 可以保持當(dāng)前裝飾器去裝飾的函數(shù)的 __name__ 的值不變

以上輸出結(jié)果就是: 

　　　　num1

　　　　num2

轉(zhuǎn)載：https://www.cnblogs.com/skaarl/p/9406910.html

總結(jié)

以上是生活随笔為你收集整理的python functiontools模块中的 wraps的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。