35 lines
1.0 KiB
Python
35 lines
1.0 KiB
Python
"""轮询 v9 执行结果"""
|
|
import time, requests, json, os, sys
|
|
from pathlib import Path
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv(Path(__file__).resolve().parents[2] / ".env")
|
|
|
|
EXEC_ID = "847822eb-e63b-46c0-929e-5d5f184a052e"
|
|
BASE = "http://localhost:8000"
|
|
TOKEN_FILE = Path(__file__).parent / ".monitor_token"
|
|
|
|
def get_token():
|
|
return TOKEN_FILE.read_text().strip()
|
|
|
|
def poll():
|
|
token = get_token()
|
|
headers = {"Authorization": f"Bearer {token}"}
|
|
for attempt in range(60):
|
|
r = requests.get(f"{BASE}/api/execution/{EXEC_ID}", headers=headers)
|
|
if r.status_code == 401:
|
|
print("Token 过期，请刷新")
|
|
sys.exit(1)
|
|
data = r.json()
|
|
status = data.get("status", "unknown")
|
|
print(f"[{attempt+1}] status={status}")
|
|
if status in ("completed", "failed", "partial"):
|
|
print(json.dumps(data, indent=2, ensure_ascii=False))
|
|
return data
|
|
time.sleep(5)
|
|
print("超时")
|
|
return None
|
|
|
|
if __name__ == "__main__":
|
|
poll()
|