feat: 报告模板系统 - prepare_report_data + generate_report + 5模板

scripts/prepare_report_data.py

@@ -0,0 +1,139 @@
+#!/usr/bin/env python3
+"""
+prepare_report_data.py — 为所有报告模板提供结构化参数值
+输出JSON，每个字段有来源标注，LLM只读不修改
+用法: python3 prepare_report_data.py [--report-type intraday|strategy|self-buy]
+"""
+
+import sqlite3, json, sys, subprocess, re, os
+from pathlib import Path
+from datetime import datetime
+
+DB = Path("/home/hmo/MoFin/data/mofin.db")
+HK_RATE_SCRIPT = Path("/home/hmo/MoFin/scripts/hk_rate.py")
+
+def get_hk_rate() -> float:
+    r = subprocess.run(["python3", str(HK_RATE_SCRIPT)], capture_output=True, text=True, timeout=10)
+    m = re.search(r"[\d.]+$", r.stdout.strip())
+    return float(m.group()) if m else 0.93
+
+def get_portfolio(db, hk_rate):
+    """总资产/现金/仓位 — 来源: portfolio_summary + holdings实时计算"""
+    r = db.execute("SELECT cash, frozen_cash FROM portfolio_summary WHERE id=1").fetchone()
+    cash, frozen = (r["cash"] or 0), (r["frozen_cash"] or 0)
+    
+    total_hkd_mv = 0.0
+    total_cny_mv = 0.0
+    holdings = []
+    for r in db.execute("""
+        SELECT code, name, shares, price, currency, market_value, cost 
+        FROM holdings WHERE is_active=1 AND (shares>0 OR shares IS NULL)
+        ORDER BY currency, code
+    """):
+        shares = r["shares"] or 0
+        if shares == 0:
+            continue
+        cur = r["currency"] or "CNY"
+        price = r["price"] or 0
+        mv = r["market_value"] or (shares * price)
+        cost_total = (r["cost"] or 0) * shares
+        pnl = mv - cost_total
+        pnl_pct = ((price / (r["cost"] or price)) - 1) * 100 if r["cost"] and r["cost"] > 0 else 0
+        
+        mv_cny = mv * hk_rate if cur == "HKD" else mv
+        if cur == "HKD":
+            total_hkd_mv += mv
+        else:
+            total_cny_mv += mv_cny
+        
+        holdings.append({
+            "code": r["code"],
+            "name": r["name"],
+            "shares": shares,
+            "price": round(price, 2),
+            "currency": cur,
+            "price_display": f"HK${price:.2f}" if cur == "HKD" else f"CNY${price:.2f}",
+            "mv": round(mv, 2),
+            "mv_display": f"HK${mv:.2f}" if cur == "HKD" else f"CNY${mv:.2f}",
+            "pnl_pct": round(pnl_pct, 2),
+            "pnl_amount": round(pnl, 2),
+        })
+    
+    stock_value_cny = total_cny_mv + total_hkd_mv * hk_rate
+    total_cash = cash + frozen
+    total_assets = stock_value_cny + total_cash
+    position_pct = (stock_value_cny / total_assets * 100) if total_assets > 0 else 0
+    
+    return {
+        "cash": round(cash, 0),
+        "frozen_cash": round(frozen, 0),
+        "total_cash": round(total_cash, 0),
+        "stock_value_cny": round(stock_value_cny, 2),
+        "total_assets": round(total_assets, 2),
+        "position_pct": round(position_pct, 1),
+        "currency": "CNY",
+        "hk_rate": hk_rate,
+        "holdings": holdings,
+        "holdings_count": len(holdings),
+        "source": "portfolio_summary + holdings实时计算",
+        "updated_at": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+    }
+
+def get_market(db):
+    """大盘数据 — 来源: market_snapshots"""
+    r = db.execute("SELECT * FROM market_snapshots ORDER BY id DESC LIMIT 1").fetchone()
+    if not r:
+        return {"error": "无市场数据"}
+    d = dict(r)
+    return {
+        "sh_index": d.get("sh_index", 0),
+        "sz_index": d.get("sz_index", 0),
+        "sh_change": d.get("sh_change_pct", 0),
+        "sz_change": d.get("sz_change_pct", 0),
+        "turnover": d.get("turnover", 0),
+        "advance_decline_ratio": d.get("advance_decline_ratio", 0),
+        "mood": d.get("mood", "unknown"),
+        "snapshot_time": d.get("snapshot_time", ""),
+        "source": "market_snapshots DB",
+    }
+
+def get_cash_history(db):
+    """现金变动历史 — 来源: cash_log"""
+    logs = []
+    for r in db.execute("SELECT * FROM cash_log ORDER BY id DESC LIMIT 5"):
+        logs.append({
+            "id": r["id"],
+            "timestamp": r["timestamp"],
+            "cash": r["cash_after"],
+            "change": r["cash_after"] - r["cash_before"],
+            "source": r["source"],
+            "note": r["note"],
+            "verified": bool(r["verified"]),
+        })
+    return logs
+
+def main():
+    report_type = "intraday"
+    if len(sys.argv) > 1 and sys.argv[1].startswith("--"):
+        report_type = sys.argv[1].split("=")[-1] if "=" in sys.argv[1] else sys.argv[1].lstrip("-")
+    
+    db = sqlite3.connect(str(DB))
+    db.row_factory = sqlite3.Row
+    hk_rate = get_hk_rate()
+    
+    data = {
+        "_meta": {
+            "report_type": report_type,
+            "generated_at": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+            "data_integrity": "所有数值由代码从DB获取，LLM不得修改",
+        },
+        "portfolio": get_portfolio(db, hk_rate),
+        "market": get_market(db),
+        "cash_history": get_cash_history(db),
+    }
+    
+    print(json.dumps(data, ensure_ascii=False, indent=2))
+    db.close()
+
+if __name__ == "__main__":
+    main()