一、上代码
# 权限表
auth_dict1 = {
1:'add用户',
2:'del用户',
3:'change用户',
4:'index用户',
}
# 用户表
user_dict = {
1: '张三',
2: '李四',
3: '王五',
4: '钱六',
}
# 职位类表
type_dict = {
1: '医生',
2: '老师',
3: '技师',
}
# 用户与职位关系表
user_type = {
user_dict[1]: type_dict[1],
user_dict[2]: type_dict[2],
user_dict[3]: type_dict[2],
user_dict[4]: type_dict[3],
}
# 职位与权限关系表
role_auth = {
type_dict[1]: [auth_dict1[1]],
type_dict[2]: [auth_dict1[1]],
type_dict[3]: [auth_dict1[1]],
}
# 打印用户与职位关系
print(user_type)
# 打印职位与权限关系
print(role_auth)
# 自定义装饰器函数
def func(fn):
# name是被装饰原始函数调用时传来的值(没有被处理的值)
def wapper(name):
print(name, '权限查询开始')
# 第一层取调用传过来人名的职位信息
job = user_type.get(name, None)
if job:
# 第二层取职位信息的权限列表,如果用户已给权限则执行原函数fn(name)
auth = role_auth.get(user_type[name], None)
if auth:
print('职业:', job)
print('权限:', auth)
# 这里的fn实际是被装饰原函数的代名称,由func(fn)参数决定fn的代表函数名
fn(name)
else:
print(f'用户{name}暂未分配权限,请分配权限后再玩')
else:
print(f'暂无该用户,请添加用户:{name}')
return wapper
@func
def dosomething(value):
# 当装饰器执行完毕时,则会执行此函数,以下就可以自定义校验后的执行逻辑
print(value, '权限查询完毕')
dosomething("张三")
二、闲言碎语
以上代码为自己对权限管理的理解,重点在于对装饰器执行逻辑的理解。只要厘清这个逻辑再数据库中建表就是顺理成章的事情了。