初始化

offline-backend-20260512/backend/scripts/migrate_roundtable_task_drafts.py

@@ -0,0 +1,127 @@
+"""一次性迁移:把已有的 taskId 会商记录从个人草稿表搬进独立的 task 存储。
+
+背景
+====
+早期 taskId 深链会商记录混存在按用户分权的 ``roundtable_drafts`` 表里(带一个
+``task_id`` 列 + 放开归属校验)。现在 taskId 聊天记录有了独立、不分权的存储
+``roundtable_task_drafts``(只按 ``task_id`` 区分,跨用户共享)。本脚本把历史数据搬过去:
+
+1. ``roundtable_drafts`` 中 ``task_id`` 非空的每一行 → 在 ``roundtable_task_drafts``
+   建同 id 的记录(``created_by`` = 原 ``user_id``,审计用);
+2. 这些草稿对应的 ``roundtable_jobs`` 行(``draft_id`` 匹配且 ``task_id`` 为空)补上
+   ``task_id``,使其转为「task 作业」(按 task 共享);
+3. 从 ``roundtable_drafts`` 删除已迁走的行,使个人历史不再混入 task 记录。
+
+幂等:目标表已存在同 id 记录则跳过;可重复运行。alembic 自动升级在运行时是关闭的
+(见 ``persistence/engine.py``),所以这步数据搬迁用本脚本手动执行。
+
+用法(在 backend 目录)
+======================
+::
+
+    # 预览(不写盘)
+    PYTHONPATH=. uv run python scripts/migrate_roundtable_task_drafts.py --dry-run
+
+    # 执行
+    PYTHONPATH=. uv run python scripts/migrate_roundtable_task_drafts.py
+"""
+
+from __future__ import annotations
+
+import argparse
+import asyncio
+import logging
+import sys
+
+logging.basicConfig(level=logging.INFO, format="%(message)s")
+logger = logging.getLogger("migrate_roundtable_task_drafts")
+
+
+async def _run(dry_run: bool) -> int:
+    from sqlalchemy import select, update
+
+    from deerflow.config import get_app_config
+    from deerflow.persistence.engine import close_engine, get_session_factory, init_engine_from_config
+    from deerflow.persistence.roundtable_drafts.model import RoundtableDraftRow
+    from deerflow.persistence.roundtable_jobs.model import RoundtableJobRow
+    from deerflow.persistence.roundtable_task_drafts.model import RoundtableTaskDraftRow
+
+    config = get_app_config()
+    await init_engine_from_config(config.database)
+    sf = get_session_factory()
+
+    moved = 0
+    skipped = 0
+    jobs_tagged = 0
+    try:
+        async with sf() as session:
+            rows = (
+                await session.execute(
+                    select(RoundtableDraftRow).where(RoundtableDraftRow.task_id.is_not(None))
+                )
+            ).scalars().all()
+            logger.info("发现 %d 条带 task_id 的个人草稿待迁移。", len(rows))
+
+            for row in rows:
+                # 幂等:目标已存在同 id → 跳过(仍尝试给作业补 task_id + 删源行)。
+                exists = await session.get(RoundtableTaskDraftRow, row.id)
+                if exists is None:
+                    if not dry_run:
+                        session.add(
+                            RoundtableTaskDraftRow(
+                                id=row.id,
+                                task_id=row.task_id,
+                                created_by=row.user_id,
+                                title=row.title or "",
+                                furthest_step=row.furthest_step or 1,
+                                step1=row.step1,
+                                step2=row.step2,
+                                step3=row.step3,
+                                created_at=row.created_at,
+                                updated_at=row.updated_at,
+                            )
+                        )
+                    moved += 1
+                    logger.info("  迁移 draft %s (task_id=%s)", row.id, row.task_id)
+                else:
+                    skipped += 1
+                    logger.info("  跳过 draft %s (目标已存在)", row.id)
+
+                # 关联后台作业补 task_id(未标记的才动)。
+                if not dry_run:
+                    res = await session.execute(
+                        update(RoundtableJobRow)
+                        .where(RoundtableJobRow.draft_id == row.id)
+                        .where(RoundtableJobRow.task_id.is_(None))
+                        .values(task_id=row.task_id)
+                    )
+                    jobs_tagged += int(res.rowcount or 0)
+
+                # 删除源行(task 记录不再留在个人表)。
+                if not dry_run:
+                    await session.delete(row)
+
+            if not dry_run:
+                await session.commit()
+
+        logger.info(
+            "完成%s:迁移 %d 条,跳过 %d 条,关联作业打标 %d 条。",
+            "(dry-run,未写盘)" if dry_run else "",
+            moved,
+            skipped,
+            jobs_tagged,
+        )
+        return 0
+    finally:
+        await close_engine()
+
+
+def main() -> int:
+    parser = argparse.ArgumentParser(description="迁移 taskId 会商记录到独立 task 存储")
+    parser.add_argument("--dry-run", action="store_true", help="只预览,不写盘")
+    args = parser.parse_args()
+    return asyncio.run(_run(args.dry_run))
+
+
+if __name__ == "__main__":
+    sys.exit(main())