假设已经存在函数 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')
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值