7635802a42
背景：系统此前没有操作审计。task_logs 的 task_id 是 NOT NULL 外键，只能挂在 任务上，且全项目仅 4 处写入点 —— 登录、导出、产品增删改、收编完全不留痕。 需求方整理的问题清单里「无审计日志查看页」正源于此：不是没有页面，是没数据。 设计参考 MOM(KCGL) 的 audit_logs / audit_listener，但按 Track 栈做了取舍： 1) 写入时机：MOM 用 SQLAlchemy event listener + 同事务写入，优点是零侵入， 缺点是**业务回滚时审计一起消失**，而失败/被拒的操作（越权尝试、参数错误） 恰恰最需要留痕。Track 改为响应生成后用**独立 session** 写入： - 业务回滚不影响审计（已验证 422/401 失败操作同样落库） - 审计写入失败也不影响业务（全包裹 try/except） - 代价：非原子提交，响应后进程立即被 kill 可能丢一条（已注释说明取舍） 2) 采集方式：中间件自动采集写操作 + 导出/下载/打印这类「读但敏感」的 GET。 路径段推导 module/action/target_id。不做手写埋点，因为手写必然漏 —— task_logs 只有 4 处写入点就是前车之鉴。 3) 增量价值：新增 request_id 字段，与 core/logging.py 的结构化日志打通， 凭一个 ID 就能从审计记录直接跳到那一次接口日志。MOM 无此字段。 4) 敏感信息：details 经 sanitize_details 递归剔除 password/token/secret 等键； 中间件不读请求体，登录明文密码不会落库（已断言表内无密码痕迹）。 配套改动： - core/roles.py：角色常量与 is_admin 收敛为单一事实来源。此前同一份 「管理员角色」规则散在 task_service、products.py 内联判断和前端 constants/task.ts 三处，已因此发生过「移动端漏判 SUPERVISOR 误挡主管」。 task_service 改为从 core.roles 导入同名常量，保持既有引用可用。 - core/deps.py：抽出 require_roles/require_admin 可复用依赖，替代内联判断。 - main.py：500 响应显式补 X-Request-ID 头 —— 该响应由 ServerErrorMiddleware 生成，位于 RequestContextMiddleware 外层，中间件没机会写头。 - auth.py：登录校验前把「尝试的账号」写入 request.state，使登录事件 （含失败登录）可归属到人，可用于追踪暴力破解。 验证：本地起 PostgreSQL 17 + 迁移后跑端到端测试，32/32 通过 （TestClient 每个请求新建事件循环，与模块级 asyncpg 连接池冲突会报 "got Future attached to a different loop"，故改用 httpx.AsyncClient + ASGITransport 单循环；生产 uvicorn 单循环无此问题）。
41 lines
1.6 KiB
Python
41 lines
1.6 KiB
Python
"""认证 API — 对接 MOM sys_user + 双 Token 刷新"""
|
||
from fastapi import APIRouter, Depends, Request
|
||
from app.schemas.user import (
|
||
LoginRequest,
|
||
LoginResponse,
|
||
RefreshRequest,
|
||
RefreshResponse,
|
||
UserResponse,
|
||
)
|
||
from app.services.auth_service import login, refresh_access_token, get_current_user
|
||
|
||
router = APIRouter(prefix="/auth", tags=["认证"])
|
||
|
||
|
||
@router.post("/login", response_model=LoginResponse)
|
||
def login_endpoint(data: LoginRequest, request: Request):
|
||
"""登录 — 验证 MOM sys_user 表，返回 Access + Refresh 双 Token"""
|
||
# 登录请求本身尚未认证，中间件拿不到操作人。但「谁在尝试登录、失败了多少次」
|
||
# 恰恰是审计里最该有的信息，所以在校验之前就把尝试的账号写进 state：
|
||
# 登录失败时同样留痕，且能按账号追踪暴力破解。
|
||
# 注意：绝不把 data.password 写进 state / 审计，密码不落库。
|
||
request.state.audit_user = data.username
|
||
return login(data.username, data.password)
|
||
|
||
|
||
@router.post("/refresh", response_model=RefreshResponse)
|
||
def refresh_endpoint(data: RefreshRequest):
|
||
"""刷新 Access Token — 使用 Refresh Token 换取新的 Access Token"""
|
||
return refresh_access_token(data.refresh_token)
|
||
|
||
|
||
@router.get("/me", response_model=UserResponse)
|
||
def get_me(current_user: dict = Depends(get_current_user)):
|
||
"""获取当前用户信息（从 Access Token 解析）"""
|
||
return UserResponse(
|
||
id=current_user["sub"],
|
||
username=current_user.get("username", ""),
|
||
display_name=current_user.get("display_name", ""),
|
||
role=current_user.get("role", "operator"),
|
||
)
|