您好,登錄后才能下訂單哦!
題意:
假設(shè)已經(jīng)存在函數(shù) get_permissions可以獲取當(dāng)前用戶的權(quán)限列表, 設(shè)計一個權(quán)限管理類,既可以作為裝飾器,對一個函數(shù)授權(quán),也可以作為上下文管理,對一段代碼授權(quán)
例如
permissions = get_permissions()
if 'admin' in permissions:
## do somthings
else:
raise Exception('Permissions denied')
分析:
權(quán)限檢查的一般都是用裝飾器的方法實現(xiàn)
demo:
def get_permissions():
return "root"
class Requrie:
def __init__(self,permissions):
self.permissions = permissions
def check(self):
if len(set(self.permissions).intersection(get_permissions())) <= 0:
raise Exception('Permissions denied')
def __call__(self,fn):
@wraps(fn)
def wrap(*args, **kwargs):
self.check()
return fn(*args, **kwargs)
return wrap
def __enter__(self):
self.check()
def __exit__(self, *args, **kwargs):
pass
def __get__(self, instance, cls): # 類作為裝飾器,一定要這樣用
type.MethodType(self, instance)
@Requrie({"admin", 'root'})
def reboot():
pass
def kill():
print("i will kill something")
with Requrie("root"):
print('kill')
免責(zé)聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點不代表本網(wǎng)站立場,如果涉及侵權(quán)請聯(lián)系站長郵箱:is@yisu.com進行舉報,并提供相關(guān)證據(jù),一經(jīng)查實,將立刻刪除涉嫌侵權(quán)內(nèi)容。