138 lines
4.4 KiB
Python
138 lines
4.4 KiB
Python
"""权限点定义与角色默认权限"""
|
|
|
|
# ---- 权限点 ----
|
|
PERMISSIONS = {
|
|
# 用户管理
|
|
"user:view": "查看用户列表",
|
|
"user:create": "创建用户",
|
|
"user:edit": "编辑用户",
|
|
"user:rename": "修改用户名",
|
|
"user:delete": "删除用户",
|
|
"user:assign_permissions": "分配权限",
|
|
# 账号管理
|
|
"account:view_all": "查看所有账号（仅用户名、标签、分配）",
|
|
"account:view_full": "查看账号完整字段（密码/邮箱，仅管理员）",
|
|
"account:view_assigned": "查看分配给自己的账号",
|
|
"account:import": "导入账号",
|
|
"account:check": "账号检测",
|
|
"account:assign": "分配账号给客服",
|
|
"account:delete": "删除账号",
|
|
# 登录任务
|
|
"login:batch": "发起批量登录",
|
|
"login:view_all": "查看所有登录任务",
|
|
"login:view_assigned": "查看自己账号的登录任务",
|
|
# Cookie
|
|
"cookie:view": "查看 Cookie",
|
|
"cookie:operate": "检测/重登已分配账号 Cookie（不含查看）",
|
|
"cookie:export": "导出 Cookie",
|
|
# 斗鱼活动
|
|
"douyu:task": "斗鱼任务管理",
|
|
"douyu:config": "斗鱼配置管理",
|
|
"yyb:session": "应用宝扫码登录与角色查询",
|
|
"yyb:recharge": "应用宝充值与付款",
|
|
"yyb:manage": "接管并操作他人的应用宝任务",
|
|
"yyb:history": "查看应用宝充值历史",
|
|
# 虎牙
|
|
"huya:account": "虎牙账号管理（兼容旧权限）",
|
|
"huya:view_all": "查看所有虎牙账号",
|
|
"huya:view_assigned": "查看分配给自己的虎牙账号",
|
|
"huya:import": "导入/登录虎牙账号",
|
|
"huya:assign": "分配虎牙账号",
|
|
"huya:delete": "删除虎牙账号",
|
|
"huya:cookie:view": "查看虎牙 Cookie",
|
|
"huya:cookie:export": "导出/清除虎牙 Cookie",
|
|
"huya:task": "虎牙任务管理",
|
|
"huya:bind": "虎牙绑定操作",
|
|
"huya:recharge": "虎牙充值操作",
|
|
"huya:config": "虎牙配置管理",
|
|
# 代理 & 白名单
|
|
"proxy:manage": "代理配置管理",
|
|
"whitelist:manage": "白名单配置管理",
|
|
"whitelist:test": "测试白名单",
|
|
# 系统
|
|
"system:settings": "系统设置",
|
|
"audit:view": "查看审计日志",
|
|
}
|
|
|
|
# ---- 角色默认权限 ----
|
|
ROLE_PERMISSIONS = {
|
|
"super_admin": list(PERMISSIONS.keys()), # 全部权限
|
|
"operation": [
|
|
"account:view_all",
|
|
"account:import",
|
|
"account:check",
|
|
"account:assign",
|
|
"login:batch",
|
|
"login:view_all",
|
|
"cookie:view",
|
|
"cookie:export",
|
|
"douyu:task",
|
|
"douyu:config",
|
|
"yyb:session",
|
|
"yyb:recharge",
|
|
"yyb:manage",
|
|
"yyb:history",
|
|
"huya:account",
|
|
"huya:view_all",
|
|
"huya:import",
|
|
"huya:assign",
|
|
"huya:delete",
|
|
"huya:cookie:view",
|
|
"huya:cookie:export",
|
|
"huya:task",
|
|
"huya:bind",
|
|
"huya:recharge",
|
|
"huya:config",
|
|
"proxy:manage",
|
|
"whitelist:manage",
|
|
"whitelist:test",
|
|
],
|
|
"support": [
|
|
"account:view_assigned",
|
|
"cookie:operate",
|
|
"douyu:task",
|
|
"yyb:session",
|
|
"yyb:history",
|
|
"huya:view_assigned",
|
|
"huya:task",
|
|
],
|
|
}
|
|
|
|
# 角色中文映射
|
|
ROLE_LABELS = {
|
|
"super_admin": "超级管理员",
|
|
"operation": "运营",
|
|
"support": "客服",
|
|
}
|
|
|
|
|
|
def get_role_permissions(role: str) -> list[str]:
|
|
"""获取角色拥有的权限列表。"""
|
|
return ROLE_PERMISSIONS.get(role, [])
|
|
|
|
|
|
def get_user_permissions(user) -> list[str]:
|
|
"""获取用户的实际权限列表。
|
|
优先使用用户自定义权限（custom_permissions），
|
|
若为 None 则回退到角色默认权限。
|
|
"""
|
|
if user.custom_permissions is not None:
|
|
return user.custom_permissions
|
|
return get_role_permissions(user.role)
|
|
|
|
|
|
def has_permission(role: str, permission: str) -> bool:
|
|
"""检查角色是否拥有某权限（仅检查角色默认权限，不考虑自定义权限）。
|
|
|
|
注意：对于需要考虑用户自定义权限的场景，请使用 user_has_permission()。
|
|
"""
|
|
return permission in get_role_permissions(role)
|
|
|
|
|
|
def user_has_permission(user, permission: str) -> bool:
|
|
"""检查用户是否拥有某权限（考虑自定义权限）。
|
|
|
|
优先使用用户的 custom_permissions，若为 None 则回退到角色默认权限。
|
|
"""
|
|
return permission in get_user_permissions(user)
|