sync: 鍚屾灞€鍩熺綉鏈€鏂颁唬鐮?(source master @ b69568e4)

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

@@ -0,0 +1,62 @@
+"""回填力量态势详情版本历史（一次脚本）。
+
+对已有 force_situation_details 行但没有任何版本记录的任务，把当前文档补写为版本 1
+（created_at 取该行的 updated_at），保证时间轴有初始刻度。
+
+用法（backend 目录）：PYTHONPATH=. uv run python scripts/backfill_force_situation_versions.py
+"""
+
+from __future__ import annotations
+
+import asyncio
+
+
+async def main() -> int:
+    from sqlalchemy import func, select
+
+    from deerflow.config import get_app_config
+    from deerflow.persistence.engine import close_engine, get_session_factory, init_engine_from_config
+    from deerflow.persistence.force_situation.model import (
+        ForceSituationDetailRow,
+        ForceSituationDetailVersionRow,
+    )
+
+    config = get_app_config()
+    await init_engine_from_config(config.database)
+    sf = get_session_factory()
+    try:
+        async with sf() as session:
+            rows = (await session.execute(select(ForceSituationDetailRow))).scalars().all()
+            patched = 0
+            for row in rows:
+                count = (
+                    await session.execute(
+                        select(func.count())
+                        .select_from(ForceSituationDetailVersionRow)
+                        .where(ForceSituationDetailVersionRow.task_id == row.task_id)
+                    )
+                ).scalar() or 0
+                if count:
+                    continue
+                session.add(
+                    ForceSituationDetailVersionRow(
+                        task_id=row.task_id,
+                        version=1,
+                        detail_json=row.detail_json,
+                        updated_by=row.updated_by,
+                        created_at=row.updated_at,
+                    )
+                )
+                patched += 1
+                print(f"回填 {row.task_id} → 版本 1")
+            await session.commit()
+            print(f"完成：回填 {patched} 个任务，跳过 {len(rows) - patched} 个（已有版本）")
+    finally:
+        await close_engine()
+    return 0
+
+
+if __name__ == "__main__":
+    import sys
+
+    sys.exit(asyncio.run(main()))