63 lines
2.5 KiB
Python
63 lines
2.5 KiB
Python
|
"""现场重跑 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()
|