bfddfdfdfewe

scripts/diag_enrich_v2.py

@@ -0,0 +1,62 @@
+"""现场重跑 enrichment_loop 查询 + 看排序"""
+import os, paramiko, base64
+c = paramiko.SSHClient()
+c.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+c.connect("207.57.129.228", port=19717, username="root",
+          password=os.environ["REMOTE_PASS"],
+          timeout=30, allow_agent=False, look_for_keys=False)
+
+py = r"""import asyncio, sys
+sys.path.insert(0, '/app')
+from sqlalchemy import select
+from app.database import AsyncSessionLocal
+from app.models.article import Article
+
+async def main():
+    async with AsyncSessionLocal() as session:
+        # 跟 enrichment_loop.py:410 一样的查询
+        rows = (await session.execute(
+            select(Article)
+            .where(Article.translation_status == "ok", Article.title_zh.is_not(None))
+            .order_by(Article.id.asc())
+            .limit(160)
+        )).scalars()
+        candidates = list(rows)
+        # 状态分布
+        cls_dist = {}
+        for a in candidates:
+            s = a.classify_status or "NULL"
+            cls_dist[s] = cls_dist.get(s, 0) + 1
+        print(f"candidates={len(candidates)}")
+        print(f"classify 分布: {cls_dist}")
+        # 头 5 篇的 id + status
+        for a in candidates[:5]:
+            print(f"  id={a.id} cls={a.classify_status} fmt={a.format_status} cmt={a.commentary_status} img={a.image_ai_status}")
+        # 尾 5 篇
+        print("--- last 5 ---")
+        for a in candidates[-5:]:
+            print(f"  id={a.id} cls={a.classify_status} fmt={a.format_status} cmt={a.commentary_status} img={a.image_ai_status}")
+        # todo 计算
+        todo = []
+        for a in candidates:
+            statuses = [a.format_status or "pending",
+                        a.classify_status or "pending",
+                        a.image_ai_status or "pending",
+                        a.commentary_status or "pending"]
+            if any(s in ("pending","failed","n/a") for s in statuses):
+                todo.append(a.id)
+            if len(todo) >= 8: break
+        print(f"todo={len(todo)} ids={todo}")
+
+asyncio.run(main())
+"""
+
+b64 = base64.b64encode(py.encode("utf-8")).decode("ascii")
+si, so, se = c.exec_command(f"bash -lc 'echo {b64} | base64 -d > /srv/news/diag.py'", timeout=10)
+so.read()
+
+# 复制进 worker 容器
+si, so, se = c.exec_command("bash -lc 'cd /srv/news && CID=$(docker compose ps -q worker) && docker cp diag.py $CID:/tmp/diag.py && docker compose exec -T worker python /tmp/diag.py 2>&1 | head -30'", timeout=30)
+print(so.read().decode(errors="replace"))
+
+c.close()