get_permissions授权获取当前用户的权限列表
题意:
假设已经存在函数 get_permissions可以获取当前用户的权限列表, 设计一个权限管理类,既可以作为装饰器,对一个函数授权,也可以作为上下文管理,对一段代码授权
例如
permissions = get_permissions()if 'admin' in permissions: ## do somthingselse: 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')
声明:本站所有文章资源内容,如无特殊说明或标注,均为采集网络资源。如若本站内容侵犯了原著者的合法权益,可联系本站删除。