python flask login_required 自定義

AssertionError: View function mapping is overwriting an existing endpoint function: main


給 flask_login 中的 login_required 裝飾器添加權(quán)限判定

def?login_required(func,authority="ANY"):

????def?wrapper():

????????@wraps(func)

????????def?decorated_view(*args,?**kwargs):

????????????if?not?current_user.is_authenticated:

????????????????return?LoginManager.unauthorized()

????????????if?current_user.WEBUserType?!=?'agent':return?{'message':?'Your?authority?is?not?valid.'},?401

????????????if?authority?!=?"customer":

????????????????if?authority?==?'cook'?and?current_user.authority?==?'customer':

????????????????????return?{'message':?'Your?authority?is?not?valid.'},?401

????????????????if?authority?==?'manager'?and?current_user.authority?!=?'manager':

????????????????????return?{'message':?'Your?authority?is?not?valid.'},?401

????????????return?func(*args,?**kwargs)

????????return?decorated_view

????wrapper.__name__?=?func.__name__?

????return?wrapper

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容