From 5bb08fe5b6974cd05e314302e45cbc5cdaada9be Mon Sep 17 00:00:00 2001 From: zlt Date: Thu, 27 Aug 2026 10:16:45 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E6=9D=83=E9=99=90?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/web/auth.py | 53 ++++++++++++---------- app/web/main.py | 18 ++++---- app/web/static/index.html | 15 ++++--- config/settings.py | 4 +- scripts/run_tests.py | 2 +- scripts/test_batch18_units.py | 83 ++++++++++++++++++----------------- 6 files changed, 92 insertions(+), 83 deletions(-) diff --git a/app/web/auth.py b/app/web/auth.py index 5a965d0..618c524 100644 --- a/app/web/auth.py +++ b/app/web/auth.py @@ -7,14 +7,11 @@ PMS 登录与权限 —— 认证解析 / 角色判定 / 会话票签验 / 接 （scripts/test_batch18_units.py）。唯一碰网络的是 call_bshop_login 一个函数， 它把真正的 HTTP 调用隔离出来，其余全部可离线测。 -三块职责： - 一、认证：解析第三方登录系统 bshop 的登录返回（parse_bshop_login）。 - 二、角色：从 bshop 返回的权限清单里认出系统管理员 / 交易员（role_from_permissions）。 - 三、会话与鉴权：自签会话票（make_session / verify_session），以及按「方法 + 路径」 - 判这个请求要什么角色（authz_decision）。 +一个人可以同时持有多个角色（既是系统管理员又是交易员），所以角色一律按「集合」处理， +不折叠成单一角色。 -会话票格式：base64url( json({phone, username, role, exp}) ) + "." + base64url( hmac_sha256 ) - —— 只用标准库 hmac / hashlib，不引新依赖。exp 是 Unix 秒。 +会话票格式：base64url( json({phone, username, roles, exp}) ) + "." + base64url( hmac_sha256 ) + —— 只用标准库 hmac / hashlib，不引新依赖。exp 是 Unix 秒。roles 是角色字符串列表。 """ from __future__ import annotations @@ -23,7 +20,7 @@ import hashlib import hmac import json import time -from typing import Optional +from typing import List, Optional # 角色常量：既是写进会话票的字面值，也是回给前端、供前端判断的字面值。 ROLE_ADMIN = "系统管理员" @@ -32,7 +29,7 @@ ROLE_TRADER = "交易员" # ============================================================ 一、认证：解析 bshop 返回 def parse_bshop_login(resp_json: dict) -> dict: - """把 bshop 登录返回解析成统一结构（不判角色，角色交给 role_from_permissions）。 + """把 bshop 登录返回解析成统一结构（不判角色，角色交给 roles_from_permissions）。 入参是已经 json 解析好的字典（真正发 HTTP 的是 call_bshop_login）。 成功: {"ok": True, "phone", "username", "permissions": [...], "orgs": [...]} @@ -83,28 +80,35 @@ def call_bshop_login(login_url: str, phone: str, password: str, timeout: int = 1 return parse_bshop_login(body) -# ============================================================ 二、角色判定 +# ============================================================ 二、角色判定（可多角色） def _markers(csv: str) -> set: """把 "pms:admin, band:yhyqx" 这类逗号分隔配置切成集合。""" return {t.strip() for t in (csv or "").split(",") if t.strip()} -def role_from_permissions(permissions, admin_perms_csv: str, - trader_perms_csv: str = "") -> Optional[str]: - """按权限清单判角色。 - - 命中任一「管理员标记」权限 -> 系统管理员 - - 否则若配置了「交易员准入」且一条都不占 -> None（无权进入） - - 否则 -> 交易员 +def roles_from_permissions(permissions, admin_perms_csv: str, + trader_perms_csv: str = "") -> List[str]: + """返回这个人持有的**全部**角色（一个人可以同时是系统管理员和交易员）。 + - 系统管理员：权限里含 PMS_ADMIN_PERMS 任一条。 + - 交易员：未配 PMS_TRADER_PERMS（留空=人人都是交易员），或权限里含其任一条。 + 两个角色都不占 -> 空列表，代表无权进入 PMS。 permissions 是 "system_key:menu_key" 字符串列表（parse_bshop_login 已归一）。 + 返回顺序固定为 [系统管理员, 交易员] 中此人占到的那些。 """ have = set(permissions or []) admin = _markers(admin_perms_csv) trader = _markers(trader_perms_csv) + roles: List[str] = [] if admin & have: - return ROLE_ADMIN - if trader and not (trader & have): - return None - return ROLE_TRADER + roles.append(ROLE_ADMIN) + if (not trader) or (trader & have): + roles.append(ROLE_TRADER) + return roles + + +def is_admin(roles) -> bool: + """会话/前端统一用这个判管理员，避免各处各写一遍。""" + return ROLE_ADMIN in (roles or []) # ============================================================ 三、会话票签验 @@ -117,7 +121,7 @@ def _b64d(s: str) -> bytes: def sign_session(payload: dict, secret: str) -> str: - """把 {phone, username, role, exp} 签成会话票字符串。""" + """把 {phone, username, roles, exp} 签成会话票字符串。""" body = json.dumps(payload, ensure_ascii=False, separators=(",", ":"), sort_keys=True).encode("utf-8") b = _b64e(body) @@ -125,12 +129,13 @@ def sign_session(payload: dict, secret: str) -> str: return b + "." + _b64e(sig) -def make_session(phone: str, username: str, role: str, secret: str, +def make_session(phone: str, username: str, roles, secret: str, ttl_hours: int, now: Optional[float] = None) -> str: - """签发一张带过期时间的会话票。now 可注入，便于单测。""" + """签发一张带过期时间的会话票。roles 是角色列表。now 可注入，便于单测。""" base = time.time() if now is None else now exp = int(base + int(ttl_hours) * 3600) - return sign_session({"phone": phone, "username": username, "role": role, "exp": exp}, secret) + return sign_session({"phone": phone, "username": username, + "roles": list(roles), "exp": exp}, secret) def verify_session(cookie: str, secret: str, now: Optional[float] = None) -> Optional[dict]: diff --git a/app/web/main.py b/app/web/main.py index ac8b194..7a8ef8c 100644 --- a/app/web/main.py +++ b/app/web/main.py @@ -71,8 +71,8 @@ async def _auth_guard(request: Request, call_next): return JSONResponse({"ok": False, "error": "未登录或登录已过期", "auth": "login_required"}, status_code=401) if (authmod.authz_decision(request.method, path) == "admin" - and sess.get("role") != authmod.ROLE_ADMIN): - return JSONResponse({"ok": False, "error": "无权限: 该操作需要交易系统管理员", "auth": "forbidden"}, + and not authmod.is_admin(sess.get("roles"))): + return JSONResponse({"ok": False, "error": "无权限: 该操作需要系统管理员", "auth": "forbidden"}, status_code=403) request.state.user = sess return await call_next(request) @@ -173,16 +173,16 @@ def api_auth_login(payload: dict = Body(default=None)): settings.PMS_BSHOP_TIMEOUT) if not res.get("ok"): return JSONResponse({"ok": False, "error": res.get("error") or "登录失败"}, status_code=401) - role = authmod.role_from_permissions(res["permissions"], settings.PMS_ADMIN_PERMS, - settings.PMS_TRADER_PERMS) - if role is None: + roles = authmod.roles_from_permissions(res["permissions"], settings.PMS_ADMIN_PERMS, + settings.PMS_TRADER_PERMS) + if not roles: return JSONResponse({"ok": False, "error": "这个账号没有持仓管理系统的使用权限"}, status_code=403) if not settings.PMS_SESSION_SECRET: return JSONResponse({"ok": False, "error": "服务未配置会话密钥 PMS_SESSION_SECRET"}, status_code=503) - token = authmod.make_session(res["phone"], res["username"], role, + token = authmod.make_session(res["phone"], res["username"], roles, settings.PMS_SESSION_SECRET, settings.PMS_SESSION_TTL_HOURS) resp = JSONResponse({"ok": True, "user": {"phone": res["phone"], "username": res["username"]}, - "role": role, "orgs": res.get("orgs") or []}) + "roles": roles, "orgs": res.get("orgs") or []}) resp.set_cookie(settings.PMS_SESSION_COOKIE, token, max_age=settings.PMS_SESSION_TTL_HOURS * 3600, httponly=True, samesite="lax", path="/") @@ -202,7 +202,7 @@ def api_me(request: Request): """当前登录人与角色。关掉登录时回一个合成的管理员, 前端就不弹登录框也不藏东西。""" if not settings.PMS_AUTH_ENABLED: return {"ok": True, "auth_enabled": False, - "user": {"phone": "-", "username": "本地(未开登录)"}, "role": authmod.ROLE_ADMIN} + "user": {"phone": "-", "username": "本地(未开登录)"}, "roles": [authmod.ROLE_ADMIN, authmod.ROLE_TRADER]} secret = settings.PMS_SESSION_SECRET sess = (authmod.verify_session(request.cookies.get(settings.PMS_SESSION_COOKIE), secret) if secret else None) @@ -210,7 +210,7 @@ def api_me(request: Request): return JSONResponse({"ok": False, "auth": "login_required"}, status_code=401) return {"ok": True, "auth_enabled": True, "user": {"phone": sess.get("phone"), "username": sess.get("username")}, - "role": sess.get("role")} + "roles": sess.get("roles") or []} @app.get("/api/overview") diff --git a/app/web/static/index.html b/app/web/static/index.html index 6c7ee4d..af5b1d3 100644 --- a/app/web/static/index.html +++ b/app/web/static/index.html @@ -432,7 +432,7 @@ body.dock-r:not(.r-fold) .side-r .strip{display:none;} {{ autoRefresh ? '自动' : '手动' }} · {{ lastRefresh }} - {{ me.username || me.phone }} · {{ me.role }} + {{ me.username || me.phone }} · {{ rolesText }} @@ -1965,8 +1965,9 @@ createApp({ const mode = ref('trader'); // ---- 登录与角色 (2026-08 加权限) const authed = ref(false), authReady = ref(false); - const me = ref({ phone: '', username: '', role: '' }); - const isAdmin = computed(() => me.value.role === '交易系统管理员'); + const me = ref({ phone: '', username: '', roles: [] }); + const isAdmin = computed(() => (me.value.roles || []).includes('系统管理员')); + const rolesText = computed(() => (me.value.roles || []).join(' · ')); const loginForm = reactive({ phone: '', password: '' }); const loginBusy = ref(false), loginErr = ref(''); _onUnauth = () => { authed.value = false; }; @@ -2404,7 +2405,7 @@ createApp({ const st = e && e.response && e.response.status; if (st === 401) { authed.value = false; return { ok: false, error: '未登录' }; } if (st === 403) { - const m = (e.response && e.response.data && e.response.data.error) || '无权限: 需要交易系统管理员'; + const m = (e.response && e.response.data && e.response.data.error) || '无权限: 需要系统管理员'; ElementPlus.ElMessage.error(m); return { ok: false, error: m }; } err.value = url + ' → ' + e.message; return { ok: false, error: e.message }; @@ -2416,7 +2417,7 @@ createApp({ try { const d = (await api.get('/api/me')).data || {}; if (d.ok) { - me.value = { phone: d.user.phone, username: d.user.username, role: d.role }; + me.value = { phone: d.user.phone, username: d.user.username, roles: d.roles || [] }; authed.value = true; if (!isAdmin.value && mode.value === 'ops') mode.value = 'trader'; } else { authed.value = false; } @@ -2440,7 +2441,7 @@ createApp({ async function doLogout() { try { await api.post('/api/auth/logout', {}); } catch (e) {} authed.value = false; - me.value = { phone: '', username: '', role: '' }; + me.value = { phone: '', username: '', roles: [] }; } async function loadParams() { @@ -3190,7 +3191,7 @@ createApp({ msgLatest, msgLevelClass, setMsgKind, msgGo, sigTab, srcOpen, toggleSrc, sigBriefHeld, sigBriefOther, lFold, rFold, railOpen, foldRail, closeRail, openRailSide, - authed, authReady, me, isAdmin, loginForm, loginBusy, loginErr, doLogin, doLogout, + authed, authReady, me, isAdmin, rolesText, loginForm, loginBusy, loginErr, doLogin, doLogout, sectClosed, toggleSect }; } }).use(ElementPlus).mount('#app'); diff --git a/config/settings.py b/config/settings.py index edab206..e2622f6 100644 --- a/config/settings.py +++ b/config/settings.py @@ -50,7 +50,7 @@ class Settings(BaseSettings): PMS_WEB_PORT: int = 38100 # ========================================================= - # 一之二、登录与权限 (PMS 自带登录 + 两个角色: 交易系统管理员 / 交易员) + # 一之二、登录与权限 (PMS 自带登录 + 两个角色: 系统管理员 / 交易员) # ========================================================= # 认证接第三方登录系统 bshop: 手机号+密码, 登录成功的返回里直接带这个人的权限清单。 # 角色从返回的 permissions 里认「管理员标记」权限得出; 会话用 PMS 自签票 (HMAC) 维持, @@ -62,7 +62,7 @@ class Settings(BaseSettings): PMS_SESSION_COOKIE: str = "pms_session" # 会话 cookie 名 PMS_BSHOP_LOGIN_URL: str = "https://bshop.bmbs.tech/authProxy/api/auth/login" # 第三方登录接口 PMS_BSHOP_TIMEOUT: int = 10 # 调 bshop 登录的超时 (秒) - PMS_ADMIN_PERMS: str = "pms:admin" # 「交易系统管理员」标记权限, "system_key:menu_key", 逗号分隔可多条 + PMS_ADMIN_PERMS: str = "pms:admin" # 「系统管理员」标记权限, "system_key:menu_key", 逗号分隔可多条 PMS_TRADER_PERMS: str = "" # 进入 PMS 至少要占的权限; 留空=任何能登录的人都算交易员 # ========================================================= diff --git a/scripts/run_tests.py b/scripts/run_tests.py index c299a8a..5ea5117 100644 --- a/scripts/run_tests.py +++ b/scripts/run_tests.py @@ -33,7 +33,7 @@ 名额/暂停买入/接力全链)/真跑写一跳摘要而预演不写/ 判分脚本聚合与对照分组 (41 例) test_batch18_units.py 登录与权限: 角色判定/会话票签验/bshop 返回解析/ - 接口鉴权(写操作默认拒绝) (14 例) + 接口鉴权(维护类归管理员) (16 例) test_wiring.py 装配自检: 服务层→核心→落表 全链路 (内存桩) (58 例) 共 560 例 任一子集失败即整体失败 (退出码 1)。 diff --git a/scripts/test_batch18_units.py b/scripts/test_batch18_units.py index a1513c2..8cd2030 100644 --- a/scripts/test_batch18_units.py +++ b/scripts/test_batch18_units.py @@ -1,15 +1,12 @@ # -*- coding: utf-8 -*- """ -第十八批: 登录与权限 (系统管理员 / 交易员) 的纯逻辑 —— 不联网、不连库 +第十八批: 登录与权限 (系统管理员 / 交易员, 可同时持有) 的纯逻辑 —— 不联网、不连库 ===================================================================== -本批只测 app/web/auth.py 里的纯函数, 不导入 app.web.main (那会拉起 fastapi 与一堆 -基础设施依赖)。覆盖四件事: - 1. 角色判定 role_from_permissions: 管理员标记命中/未中、交易员准入闸放行与拦截; - 2. 会话票 make_session / verify_session: 签发-校验往返, 篡改/过期/换密钥/空密钥一律拒; - 3. bshop 返回解析 parse_bshop_login: 喂真实样例抽出手机号/用户名/权限/组织, code!=200 判失败; - 4. 接口鉴权 authz_decision + is_public_path: 分工按前端两块视图划分 —— 交易员工作台上能点到 - 的都归交易员 (下单/撤单/拍板/策略/参数微调/大盘冷热采纳), 只有运维视图里的系统维护 - 操作 (ops/* 、ws-channel、industry 导入、人工补录提议) 归管理员; 公开白名单放行首页与登录接口。 +只测 app/web/auth.py 里的纯函数, 不导入 app.web.main。覆盖四件事: + 1. 角色判定 roles_from_permissions: 返回**角色集合**, 一个人可同时是管理员和交易员; + 2. 会话票 make_session / verify_session: roles 列表原样签发-校验, 篡改/过期/换密钥/空票一律拒; + 3. bshop 返回解析 parse_bshop_login: 喂真实样例抽字段, code!=200 判失败; + 4. 接口鉴权 authz_decision + is_public_path: 只有运维维护类接口归管理员, 其余登录人都放行。 运行: python scripts/test_batch18_units.py """ import os @@ -30,7 +27,6 @@ def case(name): return deco -# 与用户给的 bshop 登录成功返回同构的样例 (裁剪了权限条数, 字段与层级一致) SAMPLE = { "code": 200, "msg": "登录成功", "data": { @@ -49,49 +45,56 @@ SAMPLE = { } -# ---------------------------------------------------------------- 一、角色判定 -@case("角色: 命中管理员标记 => 管理员") -def _role_admin(): - r = A.role_from_permissions(["band:bkld", "pms:admin"], "pms:admin", "") - assert r == A.ROLE_ADMIN, r +# ---------------------------------------------------------------- 一、角色判定（可多角色） +@case("角色: 管理员标记 + 交易员开放 => 同时是管理员和交易员") +def _roles_both_open(): + r = A.roles_from_permissions(["band:bkld", "pms:admin"], "pms:admin", "") + assert r == [A.ROLE_ADMIN, A.ROLE_TRADER], r -@case("角色: 只有普通权限 => 交易员") -def _role_trader(): - r = A.role_from_permissions(["band:bkld", "band:yhyqx"], "pms:admin", "") - assert r == A.ROLE_TRADER, r +@case("角色: 只有普通权限 (交易员开放) => 只交易员") +def _roles_trader_only(): + r = A.roles_from_permissions(["band:bkld"], "pms:admin", "") + assert r == [A.ROLE_TRADER], r -@case("角色: 配了交易员准入且占其一 => 交易员") -def _role_trader_gate_pass(): - r = A.role_from_permissions(["band:ccgl"], "pms:admin", "band:ccgl,pms:-") - assert r == A.ROLE_TRADER, r +@case("角色: 配了交易员准入, 有管理员无交易员权限 => 只管理员") +def _roles_admin_only(): + r = A.roles_from_permissions(["pms:admin"], "pms:admin", "band:ccgl") + assert r == [A.ROLE_ADMIN], r -@case("角色: 配了交易员准入却一条不占 => None (无权进入)") -def _role_denied(): - r = A.role_from_permissions(["band:bkld"], "pms:admin", "band:ccgl,pms:-") - assert r is None, r +@case("角色: 配了交易员准入, 两者都占 => 同时两个角色 (用户场景)") +def _roles_both_gated(): + r = A.roles_from_permissions(["pms:admin", "band:ccgl"], "pms:admin", "band:ccgl") + assert r == [A.ROLE_ADMIN, A.ROLE_TRADER], r -@case("角色: 管理员标记可多条, 命中任一即算") -def _role_admin_multi(): - r = A.role_from_permissions(["band:yhyqx"], "pms:admin, band:yhyqx", "") - assert r == A.ROLE_ADMIN, r +@case("角色: 配了交易员准入却一条不占 => 空 (无权进入)") +def _roles_denied(): + r = A.roles_from_permissions(["band:x"], "pms:admin", "band:ccgl") + assert r == [], r + + +@case("角色: is_admin 辅助判定") +def _is_admin_helper(): + assert A.is_admin([A.ROLE_ADMIN, A.ROLE_TRADER]) is True + assert A.is_admin([A.ROLE_TRADER]) is False + assert A.is_admin([]) is False # ---------------------------------------------------------------- 二、会话票 -@case("会话: 签发-校验往返, payload 字段齐全") +@case("会话: roles 列表签发-校验往返, 字段齐全") def _sess_roundtrip(): - tok = A.make_session("18795373439", "老王", A.ROLE_ADMIN, "s3cr3t", 12, now=1000.0) + tok = A.make_session("18795373439", "老王", [A.ROLE_ADMIN, A.ROLE_TRADER], "s3cr3t", 12, now=1000.0) p = A.verify_session(tok, "s3cr3t", now=1001.0) assert p and p["phone"] == "18795373439" and p["username"] == "老王", p - assert p["role"] == A.ROLE_ADMIN and p["exp"] == 1000 + 12 * 3600, p + assert p["roles"] == [A.ROLE_ADMIN, A.ROLE_TRADER] and p["exp"] == 1000 + 12 * 3600, p @case("会话: 篡改正文或签名 => None") def _sess_tampered(): - tok = A.make_session("186", "u", A.ROLE_TRADER, "s3cr3t", 12, now=1000.0) + tok = A.make_session("186", "u", [A.ROLE_TRADER], "s3cr3t", 12, now=1000.0) b, sig = tok.split(".", 1) assert A.verify_session(b + "x." + sig, "s3cr3t", now=1001.0) is None assert A.verify_session(b + "." + sig + "x", "s3cr3t", now=1001.0) is None @@ -99,7 +102,7 @@ def _sess_tampered(): @case("会话: 过期 / 换密钥 / 空密钥 / 空票 一律拒") def _sess_reject(): - tok = A.make_session("186", "u", A.ROLE_TRADER, "s3cr3t", 1, now=1000.0) + tok = A.make_session("186", "u", [A.ROLE_TRADER], "s3cr3t", 1, now=1000.0) assert A.verify_session(tok, "s3cr3t", now=1000.0 + 3600 + 1) is None, "过期没拒" assert A.verify_session(tok, "other", now=1001.0) is None, "换密钥没拒" assert A.verify_session(tok, "", now=1001.0) is None, "空密钥没拒" @@ -117,11 +120,11 @@ def _parse_ok(): assert len(r["orgs"]) == 2 and r["orgs"][0]["org_name"] == "高姐", r["orgs"] -@case("解析: 样例接角色 —— 默认标记判交易员, 复用 band:yhyqx 判管理员") -def _parse_then_role(): +@case("解析: 样例接角色 —— 默认标记只交易员, band:yhyqx 标记则管理员+交易员") +def _parse_then_roles(): r = A.parse_bshop_login(SAMPLE) - assert A.role_from_permissions(r["permissions"], "pms:admin", "") == A.ROLE_TRADER - assert A.role_from_permissions(r["permissions"], "band:yhyqx", "") == A.ROLE_ADMIN + assert A.roles_from_permissions(r["permissions"], "pms:admin", "") == [A.ROLE_TRADER] + assert A.roles_from_permissions(r["permissions"], "band:yhyqx", "") == [A.ROLE_ADMIN, A.ROLE_TRADER] @case("解析: code!=200 / 缺手机号 / 非字典 => 失败")