# -*- coding: utf-8 -*-
"""轮询第八次执行结果。启动前需先运行 resubmit_v8.py 获取 execution_id 并填入下方。"""
import json
import sys
import time
from pathlib import Path

import requests
from dotenv import load_dotenv

load_dotenv(Path(__file__).resolve().parents[2] / ".env")

TOKEN_FILE = Path(__file__).parent / ".monitor_token"
BASE = "http://localhost:8000"
EXECUTION_ID = "f943bac6-23be-45c5-8b8c-a864e85a1916"
REFRESH_TOKEN = (
    "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9."
    "eyJzdWIiOiIxIiwic2l0ZV9pZCI6Mjc5MDY4NTQxNTQ0MzI2OSwidHlwZSI6InJlZnJlc2giLCJleHAiOjE3NzIyNjM0NjN9."
    "XYoda5lfxNtTSAGWoLlYhS9cA-hTK9iqK0SqUyn2KV4"
)

if EXECUTION_ID == "__FILL_ME__":
    print("请先填入 execution_id")
    sys.exit(1)


def refresh_token() -> str:
    resp = requests.post(f"{BASE}/api/auth/refresh", json={"refresh_token": REFRESH_TOKEN}, timeout=10)
    if resp.status_code != 200:
        print(f"刷新失败: {resp.status_code}")
        sys.exit(1)
    t = resp.json()["access_token"]
    TOKEN_FILE.write_text(t, encoding="utf-8")
    return t


TOKEN = refresh_token()
HEADERS = {"Authorization": f"Bearer {TOKEN}", "Content-Type": "application/json"}

from _env_paths import get_output_path

for i in range(90):
    time.sleep(20)
    mm, ss = divmod((i + 1) * 20, 60)
    try:
        r = requests.get(f"{BASE}/api/execution/history?limit=5", headers=HEADERS, timeout=15)
        if r.status_code == 401:
            TOKEN = refresh_token()
            HEADERS["Authorization"] = f"Bearer {TOKEN}"
            continue
        if r.status_code != 200:
            continue
        target = next((h for h in r.json() if h["id"] == EXECUTION_ID), None)
        if not target:
            print(f"[{mm}m{ss}s] 等待...")
            continue
        status = target.get("status")
        dur = target.get("duration_ms")
        dur_s = f"{dur/1000:.1f}s" if dur else "—"
        if status in ("success", "failed", "cancelled"):
            print(f"完成 status={status}, 耗时={dur_s}, exit_code={target.get('exit_code')}")
            lr = requests.get(f"{BASE}/api/execution/{EXECUTION_ID}/logs", headers=HEADERS, timeout=30)
            if lr.status_code == 200:
                ld = lr.json()
                out = get_output_path("SYSTEM_LOG_ROOT")
                (out / "2026-02-21__etl_run_raw_v8.json").write_text(
                    json.dumps(ld, ensure_ascii=False, indent=2), encoding="utf-8"
                )
                print("日志已保存")
                el = (ld.get("error_log") or "").strip().split("\n")
                print(f"--- error_log 末尾 80 行 (共 {len(el)} 行) ---")
                for line in el[-80:]:
                    print(line)
            break
        print(f"[{mm}m{ss}s] status={status}")
    except Exception as e:
        print(f"[{mm}m{ss}s] {e}")
else:
    print("超时")