xsl
e8208d2972
POST /reports 骨架：资格校验→终身配额预扣→tk_task→cpu 队列；动态装配 流水线（dotted-path 注册、缺阶段即 CONFIG_ERROR）；中间态逐段落库、失败 退款、derived_features 跨重试/重生成复用（T-P3-07/09）。 实现：packages/report 的 pipeline.py（Stage 契约保持不变，新增 assemble_stages/ run_pipeline/StageFailureError，ReportContext 增可选 task_id 参数、向后兼容）、 repo.py、schemas.py、service.py（重放优先于资格校验、reserve+建行+建任务同事务、 提交后才投递、并发同键落败回滚）；apps/api/routers/reports.py 填桩（send_task 按 任务名投递，不 import apps.worker）；apps/worker/report_tasks.py（worker 侧自建 gateway 装配、状态归一 QUEUED→RUNNING、成功 commit 配额+成本聚合写 model_info、 基础设施异常重试一次、重试耗尽兜底失败退款）。测试：单测 6 个（装配缺失/空配置/ 协议违约/顺序执行/失败保留产物），集成 6 场景（预扣 lifetime/PENDING、成功 committed+两阶段键、同 request_key 幂等不重复扣、二次首生成 40920、阶段失败 refund+derived_features 保留、regenerate 走 m: 月配额 version+1 复用视觉中间态）。 偏差说明： 1) apps/worker/celery_app.py 的 include 只含 apps.worker.tasks（禁改文件），真 worker 进程注册 meida.report.generate 需协调者合并时把 include 补上 apps.worker.report_tasks；API 侧已按任务名 send_task，两侧只靠名字衔接，测试 不受影响。2) ReportContext 增加可选 task_id 关键字参数（失败结案要推动 tk_task），Stage 协议/DEFAULT_STAGE_PATHS/stage_names 未动，T16/T17 兼容。 3) JSONB 深变更用 flag_modified 显式标脏（重新赋相等新 dict 会被 flush 的按值 比较跳过，集成测试抓到过该坑）。4) POST 后 tk_task 保持 PENDING（任务文档 §4 场景 1 口径），QUEUED→RUNNING 由 worker 入口归一。
48 lines
1.6 KiB
Python
48 lines
1.6 KiB
Python
"""rp_report / rp_input 的数据访问。**只有本模块可以触这两张表**（rp_share 归后续任务）。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from packages.shared.models import RpInput, RpReport
|
|
|
|
STATUS_SUCCEEDED = "succeeded"
|
|
TIER_BASIC = "basic"
|
|
|
|
|
|
async def get_report(session: AsyncSession, report_id: int) -> RpReport | None:
|
|
return await session.get(RpReport, report_id)
|
|
|
|
|
|
async def find_succeeded_basic(session: AsyncSession, *, user_id: int) -> int | None:
|
|
"""首次终身资格：该用户是否已有 succeeded 的 basic 报告（需求 §2.1）。
|
|
|
|
只取 report_id 不取整行——资格判定不需要报告内容。
|
|
"""
|
|
result = await session.execute(
|
|
select(RpReport.report_id)
|
|
.where(
|
|
RpReport.user_id == user_id,
|
|
RpReport.tier == TIER_BASIC,
|
|
RpReport.status == STATUS_SUCCEEDED,
|
|
)
|
|
.limit(1)
|
|
)
|
|
return result.scalar_one_or_none()
|
|
|
|
|
|
async def list_reports(session: AsyncSession, *, user_id: int, limit: int = 20) -> list[RpReport]:
|
|
result = await session.execute(
|
|
select(RpReport)
|
|
.where(RpReport.user_id == user_id)
|
|
.order_by(RpReport.created_at.desc(), RpReport.report_id.desc())
|
|
.limit(limit)
|
|
)
|
|
return list(result.scalars().all())
|
|
|
|
|
|
async def get_input(session: AsyncSession, report_id: int) -> RpInput | None:
|
|
result = await session.execute(select(RpInput).where(RpInput.report_id == report_id))
|
|
return result.scalar_one_or_none()
|