#!/usr/bin/env python3
"""refresh_macro_context.py — 刷新macro_context_log

读取最新market_snapshots + 腾讯实时指数, 写入macro_context_log
让load_macro_context()拿到最新市场偏向, 而不是12天前的数据

每30分钟跑一次（交易日）
"""
import json, sqlite3, urllib.request, re, sys
from pathlib import Path
from datetime import datetime

DB = Path("/home/hmo/MoFin/data/mofin.db")

def fetch_index(code, name):
    """从腾讯API拿指数行情"""
    try:
        url = f"http://qt.gtimg.cn/q={code}"
        req = urllib.request.Request(url, headers={"User-Agent": "Mozilla/5.0"})
        resp = urllib.request.urlopen(req, timeout=5)
        text = resp.read().decode("gbk")
        m = re.search(r'~([^~]*)~([^~]*)~([\d.]+)~([\d.]+)~([\d.]+)~([\d.]+)', text)
        if m:
            _, _, price, prev_close, open_p, high_low = m.groups()
            high = high_low.split("~")[0] if "~" in high_low else high_low[:8]
            low = high_low.split("~")[1] if "~" in high_low else "0"
            change_pct = (float(price) - float(prev_close)) / float(prev_close) * 100 if float(prev_close) else 0
            return {
                "price": float(price),
                "change_pct": round(change_pct, 2),
                "high": float(high) if high else float(price),
                "low": float(low) if low else float(price)
            }
    except:
        return None

def main():
    # 采集各指数
    indices = {
        "上证指数": fetch_index("sh000001", "上证指数"),
        "深证成指": fetch_index("sz399001", "深证成指"),
        "创业板指": fetch_index("sz399006", "创业板指"),
        "恒生指数": fetch_index("szHSI", "恒生指数"),
        "国企指数": fetch_index("szHSCEI", "国企指数"),
    }
    indices = {k: v for k, v in indices.items() if v}

    # 计算偏向
    sh = indices.get("上证指数", {})
    sh_change = sh.get("change_pct", 0) if sh else 0
    if sh_change < -1.5:
        overall = "bearish"
        desc = "大盘偏弱"
    elif sh_change > 1.0:
        overall = "bullish"
        desc = "大盘偏强"
    else:
        overall = "neutral"
        desc = "大盘震荡"

    structure = json.dumps({"overall": overall, "description": desc}, ensure_ascii=False)
    indices_json = json.dumps(indices, ensure_ascii=False)

    # 写入macro_context_log
    now = datetime.now()
    session = "midday" if now.hour >= 12 else "morning"
    ts = now.strftime("%Y-%m-%d %H:%M:%S")
    try:
        conn = sqlite3.connect(str(DB))
        conn.execute("""
            INSERT INTO macro_context_log 
            (data_timestamp, session, has_valid_data, indices, structure, created_at)
            VALUES (?, ?, 1, ?, ?, ?)
        """, (ts, session, indices_json, structure, ts))
        conn.commit()
        conn.close()
        print(f"[SILENT] macro_context updated: {overall} {desc} {len(indices)} indices")
    except Exception as e:
        print(f"[SILENT] macro_context write failed: {e}")

if __name__ == "__main__":
    main()