"""pull + 重建 worker + 扫描 DB 把翻译失败的改回 pending + 看新 worker 跑起来。"""
import os, paramiko, json, time
PW = os.environ["REMOTE_PASS"]
c = paramiko.SSHClient()
c.set_missing_host_key_policy(paramiko.AutoAddPolicy())
c.connect("207.57.129.228", port=19717, username="root", password=PW, timeout=15, allow_agent=False, look_for_keys=False)
def run(cmd, t=60):
    si, so, se = c.exec_command(cmd, timeout=t)
    out = so.read().decode("utf-8", "replace")
    err = se.read().decode("utf-8", "replace")
    rc = so.channel.recv_exit_status()
    if out: print(out, end="")
    if err and "Warning" not in err and "warn" not in err: print(err, end="", file=__import__("sys").stderr)
    return out

# 1) pull
print("--- pull ---")
run("cd /srv/news && sudo -u news git pull --rebase 2>&1 | tail -3")

# 2) 重建 worker
print("--- 重建 worker ---")
run("cd /srv/news && docker compose up -d --force-recreate --no-deps --build worker 2>&1 | tail -5", t=120)
time.sleep(5)

# 3) 扫描 DB:title_zh/body_zh_text 含 '翻译失败' 改回 pending
print("--- DB 扫描 ---")
out = run("docker exec news-aggregator-postgres-1 psql -U news -d news -tA -c \"SELECT count(*) FROM articles WHERE title_zh LIKE '%[翻译失败:%' OR body_zh_text LIKE '%[翻译失败:%' OR body_zh_html LIKE '%[翻译失败:%';\"")
print(f"  含 '翻译失败' 占位符的文章数: {out.strip()}")
n = run("docker exec news-aggregator-postgres-1 psql -U news -d news -tA -c \"UPDATE articles SET translation_status='pending', title_zh=NULL, body_zh_text=NULL, body_zh_html=NULL, translated_at=NULL, translation_engine=NULL, translation_chars=0 WHERE title_zh LIKE '%[翻译失败:%' OR body_zh_text LIKE '%[翻译失败:%' OR body_zh_html LIKE '%[翻译失败:%';\"")
print(f"  UPDATE 状态: {n.strip()}")

# 4) 看 worker 是否在跑 translation_loop
print("\n--- worker 日志(看 translation_loop 启动 + 节奏)---")
time.sleep(15)
out = run("docker logs --tail=50 news-aggregator-worker-1 2>&1 | tail -30", t=15)
print(out)

# 5) 等 30 秒再看(应该已经翻译 30 篇左右)
print("\n--- 等 30 秒看翻译进度 ---")
time.sleep(30)
out = run("docker exec news-aggregator-postgres-1 psql -U news -d news -c \"SELECT translation_status, count(*) FROM articles GROUP BY 1 ORDER BY 1;\"")
print(out)

# 6) redis usage
rpw = run("grep ^REDIS_PASSWORD /srv/news/.env | cut -d= -f2").strip()
out = run("docker exec news-aggregator-redis-1 redis-cli -a '" + rpw + "' GET translation:month:202606 2>/dev/null")
print(f"\n--- redis usage: {out.strip()}")

# 7) 验证 fetch_one_source 不再自动翻译(看 worker 日志确认)
print("\n--- worker 进程信息 ---")
run("docker ps --filter 'name=news-aggregator-worker' --format 'table {{.Names}}\\t{{.Status}}'")
c.close()