python - flask 用戶權(quán)限修飾器
問(wèn)題描述
看書(shū)發(fā)現(xiàn)這段代碼有點(diǎn)云里霧里,裝飾器有點(diǎn)昏,希望幫我解析下這段代碼!!
from functools import wrapsfrom flask import abortfrom flask_login import current_userfrom .models import Permissiondef permission_required(permission): def decorator(f):@wraps(f)def decorated_function(*args, **kwargs): if not current_user.can(permission):abort(403) return f(*args, **kwargs)return decorated_function return decoratordef admin_required(f): return permission_required(Permission.ADMINISTER)(f)
問(wèn)題解答
回答1:首先你要理解裝飾器的原理:
@abcddef f(): pass
實(shí)際上與下面的語(yǔ)句等價(jià):
def f(): passf=abcd(f)
我們現(xiàn)在有一個(gè)函數(shù)abcd,這個(gè)函數(shù)的本質(zhì)是:接受另一個(gè)函數(shù)當(dāng)做參數(shù),且返回一個(gè)函數(shù)。(至于返回的函數(shù)用來(lái)干嘛那就是你的事了)。這時(shí)候,abcd僅僅就是個(gè)函數(shù)而已,還不是修飾器。而由于以下這個(gè)需求十分的常見(jiàn):有一個(gè)舊函數(shù),我們又想定義一個(gè)新函數(shù),這個(gè)新函數(shù)大體上功能與舊函數(shù)接近,只是多了一點(diǎn)點(diǎn)的新功能,比如打印個(gè)日期,判斷個(gè)權(quán)限什么的。那么定義新函數(shù)的過(guò)程中肯定會(huì)調(diào)用這個(gè)舊函數(shù),然而新函數(shù)其實(shí)改變不大,舊函數(shù)往往也沒(méi)用了(因?yàn)橐话阄覀兒竺娑际怯玫男潞瘮?shù)了),那么為了不讓命名空間變得混亂和方便開(kāi)發(fā),我們可以簡(jiǎn)單的就用舊函數(shù)的名字來(lái)表示新函數(shù),也就是在定義完了一個(gè)新函數(shù)之后,我們把它的名字又變回以前的f,而以前的f就不要了。所以我們可以這樣做:定義一個(gè)函數(shù)abcd,它接受一個(gè)函數(shù)f,且返回一個(gè)新的函數(shù),再把它的返回值(新函數(shù)),再賦值給f(python里函數(shù)名也可以賦值,成為另一個(gè)函數(shù))。這實(shí)際上就是我上面的第二段代碼做的事情。由于這個(gè)需求太過(guò)常見(jiàn),所以python專門(mén)為它定義了語(yǔ)法。你不是每次都要f=abcd(f)嗎,那你就直接在f的def語(yǔ)句前面加個(gè)@abcd得了,也別每次再寫(xiě)后面那句了,不僅麻煩,有時(shí)還容易誤解。這時(shí)候,abcd就成為了裝飾器。了解了這個(gè)等價(jià)關(guān)系,你的函數(shù)就好理解了:當(dāng)你在某處使用的時(shí)候,是這樣的
@permission_required(permission)def old(): pass
等價(jià)于
def old(): passold = permission_required(permission)(old)
優(yōu)先級(jí)相同,運(yùn)算從左到右,首先計(jì)算permission_required(permission),它返回decorator是一個(gè)函數(shù),這時(shí)候變成old = decorator(old)為了滿足成為修飾器的要求,這個(gè)decorator應(yīng)當(dāng)返回一個(gè)新函數(shù)才行,在這里就是decorated_function,所以原賦值語(yǔ)句變成old = decorated_function。到這里比較清晰了,把一個(gè)函數(shù)的名字賦值給一個(gè)變量(old),所以old就變成了decorated_function這里所定義的函數(shù)。過(guò)程也就是:
old = permission_required(permission)(old)-> old = decorator(old)-> old = decorated_function回答2:
希望下面代碼對(duì)你有幫助
from functools import wrapsdef permission_required(permission): '''返回裝飾器,裝飾器中使用入?yún)?permission ''' def decorator(f):@wraps(f)def decorated_function(*args, **kwargs): if not permission:print ’403’return return f(*args, **kwargs)return decorated_function return decoratordef admin_required_true(f): '''裝飾器函數(shù),返回裝飾器 ''' return permission_required(True)(f)def admin_required_false(f): '''裝飾器函數(shù),返回裝飾器 ''' return permission_required(False)(f)@admin_required_truedef foo(): '''使用裝飾器 ''' print ’foo’ @admin_required_falsedef bar(): '''使用裝飾器 ''' print ’bar’foo()bar()
運(yùn)行結(jié)果:
foo403
相關(guān)文章:
1. javascript - js 有什么優(yōu)雅的辦法實(shí)現(xiàn)在同時(shí)打開(kāi)的兩個(gè)標(biāo)簽頁(yè)間相互通信?2. css3 - Typecho 后臺(tái)部分表單按鈕在 Chrome 下出現(xiàn)靈異動(dòng)畫(huà)問(wèn)題,求解決3. javascript - angular和jquery都用到了$符號(hào),一起用會(huì)不會(huì)沖突?4. javascript - 怎樣限制同一個(gè)瀏覽器不能登錄兩個(gè)賬號(hào)5. 想找個(gè)php大神仿個(gè)網(wǎng)站。6. java - android代碼重構(gòu):如何把a(bǔ)pp設(shè)置里的頭像UI做成通用的?7. java - 新手做一個(gè)安卓視頻播放器,想實(shí)現(xiàn)一個(gè)進(jìn)度條,按鈕那種在視頻下方懸浮的功能,不知道思路!8. javascript - jquery怎么給select option一個(gè)點(diǎn)擊時(shí)觸發(fā)的事件,如圖 如果選擇自定義觸發(fā)一個(gè)時(shí)間?9. nginx配置server模塊的問(wèn)題10. mysql優(yōu)化 - 關(guān)于mysql分區(qū)
