您好,登錄后才能下訂單哦!
題意:
假設已經存在函數 get_permissions可以獲取當前用戶的權限列表, 設計一個權限管理類,既可以作為裝飾器,對一個函數授權,也可以作為上下文管理,對一段代碼授權
例如
permissions = get_permissions()
if 'admin' in permissions:
## do somthings
else:
raise Exception('Permissions denied')
分析:
權限檢查的一般都是用裝飾器的方法實現
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')
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。