淺析裝飾器的那些事兒
點(diǎn)擊上方“IT共享之家”,進(jìn)行關(guān)注
回復(fù)“資料”可獲贈(zèng)Python學(xué)習(xí)福利
一、裝飾器的簡(jiǎn)單定義
外層函數(shù)返回里層函數(shù)的引用,里層函數(shù)引用外層函數(shù)的變量。
二、裝飾器的作用
通俗來講裝飾器的作用就是在不改變已有函數(shù)代碼前提下,為該函數(shù)增加新的功能。
def run():print('我會(huì)跑')fun()
現(xiàn)在我想在原有函數(shù)的基礎(chǔ)上新增一個(gè)功能:我會(huì)唱歌。這個(gè)時(shí)候利用裝飾器則輕松可以幫我們實(shí)現(xiàn)這個(gè)功能。
三、實(shí)例理解
(1)不傳參的裝飾器
def outer(fun):def inner():fun() //fun是外層函數(shù)的變量,在inner里面用return inner //inner就是里層函數(shù)的引用
(2)傳遞參數(shù)的裝飾器:
def func(fun):def add(*args,**kwarge):return fun(*args,**kwargs)return add
現(xiàn)在對(duì)于裝飾器的基本格式有一定的了解,就可以直接寫函數(shù)了。下面實(shí)現(xiàn)文章開頭的 我會(huì)唱歌 的功能
def outer(fun):def inner(*args, **kwarge):print("我會(huì)唱歌")return fun(*args, **kwarge)return inner
四、如何使用裝飾器
方法一:使用@符號(hào)+裝飾器的名字 把它放在想要裝飾函數(shù)的上一行即可def run():print('我會(huì)跑')run()方法二:def run():print('我會(huì)跑')run=outer(run) #就等價(jià)于@outerrun()最終打印結(jié)果是:我會(huì)唱歌我會(huì)跑
如果我想知道fun 傳遞的參數(shù)是什么,在裝飾器內(nèi)部可以使用如下方式:
def outer(fun):a = 1def inner(*args, **kwarge): # args是一個(gè)數(shù)組,kwargs一個(gè)字典print(fun.__name__) #打印fun接收的函數(shù)的名字print("我會(huì)唱歌")return fun(*args, **kwarge)return inner
但是如果我們 print(run.__name__,6666666) 輸出的結(jié)果是inner,并不是我們想要的run,這里的函數(shù)被warpTheFunction替代了。它重寫了我們函數(shù)的名字和注釋文檔(docstring)。解決方法如下:
from functools import wrapsdef outer(fun):def inner(*args, **kwargs):print(fun.__name__,11111111111)print("我會(huì)唱歌")return fun(*args, **kwargs)return innerdef run():print('我會(huì)跑')print(run.__name__,6666666) //輸出結(jié)果為 run 666666
?
五、自己實(shí)現(xiàn)裝飾器
def subuser_keymanage(view_func):'''功能是實(shí)現(xiàn)用戶管理權(quán)限的判定'''def _wrapper_view(request, *args, **kwargs):user = request.user #一個(gè)Customer對(duì)象,包含了用戶名/密碼等信息customer = user.customer.customer_id #用戶的idselect_status = get_curuser_permission(user=user, customer=customer)#調(diào)用函數(shù)返回的值有兩種0和1if not select_status:#如果返回0表示沒有權(quán)限,返回錯(cuò)誤碼return render_response(request, ErrorCode.FAILED)return view_func(request, *args, **kwargs)return _wrapper_view
@subuser_keymanagedef generate_subuser_ak_sk(request):params = json.loads(request.body)user_id_only = params.get("user_id")中間代碼就忽略了......return render_response(request, ErrorCode.FAILED)
六、裝飾器小結(jié)
通過裝飾器很大程度上可以減少代碼的復(fù)用,在代碼規(guī)范中這一點(diǎn)是很重要的。
以上就是裝飾器的基本知識(shí),即便沒有任何基礎(chǔ),按照作者的思路,套用固定的格式,不需要完全理解,只要按照流程一步一步就能寫出高端大氣上檔次的裝飾器了,恭喜你!
前方高能請(qǐng)注意:裝飾器傳參,三層嵌套函數(shù)一般用的比較少,其實(shí)也不難,一層一層看,跟上文講的一樣,僅作為知識(shí)的拓寬。
import loggingdef use_logging(level):def decorator(func):def wrapper(*args, **kwargs):if level == "warn":logging.warn("%s is running" % func.__name__)elif level == "info":logging.info("%s is running" % func.__name__)return func(*args)return wrapperreturn decoratordef foo(name='foo'):print("i am %s" % name)foo()
i am fooWARNING:root:foo is running
如果在操作過程中有任何問題,記得下面留言,我們看到會(huì)第一時(shí)間解決問題。
看完本文有收獲?請(qǐng)轉(zhuǎn)發(fā)分享給更多的人
IT共享之家
入群請(qǐng)?jiān)谖⑿藕笈_(tái)回復(fù)【入群】
-------------------?End?-------------------
往期精彩文章推薦:
