yml2213
f745a61df1
后台原本靠人工逐批点「CK 检测」发现失效账号，1.2 万个账号根本盯不过来。
P1a 把「哪些账号失效了」自动化：自动挑久未检测的账号、调现成检测接口、
把结果写回 ck_check_*。**本作业不替换任何 CK**，重登仍由人工点，符合规划里
「先做只读/只提示，确认无误再做自动动作」的渐进上线原则。
实现：
- 迁移 20261002_0037：automation_jobs / automation_runs 两张表，
login_tasks 加 relogin_attempts / relogin_next_at / relogin_last_error
（为 P1b 备好），accounts 加 auto_managed（特殊账号不参与自动化）；
- services/automation/scheduler.py：进程内调度线程，60s tick，状态全部落库
（重启不丢调度进度、不重复跑轮）；AUTOMATION_ENABLED 默认关闭，作业开关
也默认关闭；ensure_jobs 只补缺失行、不覆盖运维改过的配置；
作业异常只记 failed，不拖垮调度线程；单容器 --workers 1，无需分布式锁；
- services/automation/jobs/ck_health.py：选行口径与人工检测一致
（cookie_holding_filter：持有 CK 且无在途尝试），叠加久未检测、
auto_managed、batch_size，按 ck_checked_at 升序（最久没查的优先）；
只写 ck_check_status / ck_check_result / ck_checked_at；
- routers/automation.py：GET /jobs、PATCH /jobs/{key}、POST /jobs/{key}/run、
GET /runs；新增权限点 automation:view（运营可看）与 automation:manage
（仅超管），改配置与手动触发都写审计；
- 顺带补 POST /api/login/batch 的审计（此前批次是谁发起、多少账号无据可查）；
- 前端 /automation 页：作业卡片（开关/周期/单轮条数/并发）、运行历史、
「立即跑一次」；总开关未打开、调度线程未运行时给明确提示。
验证：
- 22 条新测试覆盖调度器行为（只跑到点的、异常不抛出、下轮时间、配置不被重置）、
选行条件（在途尝试/auto_managed/间隔/batch_size）、接口与权限，其中一条
专门断言巡检后 cookie/status/message/attempt_status 一字未动；
- 全量 257 条通过；前端 tsc 与构建通过；
- 端到端用 3 个真实账号跑真实斗鱼接口：巡检结果与直接调 check_douyu_cookie
逐条一致（3/3 valid），CK 与尝试状态全程未变。
顺带修掉一处会炸生产的坑：DB 时间列读出来是 naive UTC，直接与
datetime.now(UTC) 比较会 TypeError，现统一走 _as_utc() 补时区。
148 lines
4.7 KiB
Python
148 lines
4.7 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:tag": "修改账号标签",
|
|
"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:tag": "修改虎牙账号标签",
|
|
"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": "测试白名单",
|
|
# 自动化
|
|
"automation:view": "查看自动化作业与运行历史",
|
|
"automation:manage": "修改自动化配置、手动触发作业",
|
|
# 系统
|
|
"system:settings": "系统设置",
|
|
"audit:view": "查看审计日志",
|
|
}
|
|
|
|
# ---- 角色默认权限 ----
|
|
ROLE_PERMISSIONS = {
|
|
"super_admin": list(PERMISSIONS.keys()), # 全部权限
|
|
"operation": [
|
|
"account:view_all",
|
|
"account:import",
|
|
"account:tag",
|
|
"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:tag",
|
|
"huya:assign",
|
|
"huya:delete",
|
|
"huya:cookie:view",
|
|
"huya:cookie:export",
|
|
"huya:task",
|
|
"huya:bind",
|
|
"huya:recharge",
|
|
"huya:config",
|
|
"proxy:manage",
|
|
"whitelist:manage",
|
|
"whitelist:test",
|
|
"automation:view",
|
|
],
|
|
"support": [
|
|
"account:view_assigned",
|
|
"account:tag",
|
|
"cookie:operate",
|
|
"douyu:task",
|
|
"yyb:session",
|
|
"yyb:history",
|
|
"huya:view_assigned",
|
|
"huya:tag",
|
|
"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)
|