"""市场概览 API""" from datetime import datetime from fastapi import APIRouter, Depends from app.data.tushare_client import tushare_client from app.data import tencent_client from app.engine.recommender import get_latest_recommendations from app.config import is_trading_hours, is_market_session, should_prefer_realtime_today from app.core.deps import get_current_admin router = APIRouter(prefix="/api/market", tags=["market"]) @router.get("/temperature") async def get_temperature(): """获取市场温度（只读缓存，不触发扫描）""" result = await get_latest_recommendations() mt = result.get("market_temp") if mt: return { "trade_date": mt.trade_date, "temperature": mt.temperature, "up_count": mt.up_count, "down_count": mt.down_count, "limit_up_count": mt.limit_up_count, "limit_down_count": mt.limit_down_count, "max_streak": mt.max_streak, "broken_rate": mt.broken_rate, "index_above_ma20": getattr(mt, "index_above_ma20", False), "is_trading": is_trading_hours(), } return { "trade_date": "", "temperature": 0, "up_count": 0, "down_count": 0, "limit_up_count": 0, "limit_down_count": 0, "max_streak": 0, "broken_rate": 0, "index_above_ma20": False, "is_trading": is_trading_hours(), } @router.get("/overview") async def get_overview(): """市场概况：上证、深证、创业板指数 盘中用腾讯实时行情，盘后用 Tushare 日线（有缓存）。 """ latest_trade_date = tushare_client.get_latest_trade_date() if should_prefer_realtime_today(latest_trade_date): return await _overview_realtime() return _overview_daily() @router.get("/strategy-board") async def get_strategy_board(): """获取今日市场作战面板（只读，不触发 LLM）""" from app.llm.strategy_board import build_strategy_board return await build_strategy_board(include_llm=False) @router.get("/strategy-iteration") async def get_strategy_iteration(limit: int = 50): """获取策略复盘迭代建议（只读，不触发 LLM）""" from app.llm.strategy_iteration import build_strategy_iteration_report return await build_strategy_iteration_report(limit=limit, include_llm=False) @router.get("/ops-status") async def get_ops_status(): """管理员任务中心状态与数据新鲜度（只读，不触发扫描或 LLM）。""" from sqlalchemy import text from app.db.database import get_db from app.engine.recommender import _scan_running async with get_db() as db: rec_row = (await db.execute( text( "SELECT created_at FROM recommendations " "ORDER BY created_at DESC LIMIT 1" ) )).fetchone() tracking_row = (await db.execute( text( "SELECT track_date, created_at FROM recommendation_tracking " "ORDER BY track_date DESC, id DESC LIMIT 1" ) )).fetchone() market_row = (await db.execute( text( "SELECT trade_date, created_at FROM market_temperature " "ORDER BY REPLACE(trade_date, '-', '') DESC, id DESC LIMIT 1" ) )).fetchone() sector_row = (await db.execute( text( "SELECT trade_date, created_at FROM sector_heat " "ORDER BY REPLACE(trade_date, '-', '') DESC, id DESC LIMIT 1" ) )).fetchone() def _fmt_dt(value): return str(value or "") latest_market_date = str(market_row._mapping["trade_date"]) if market_row else "" latest_sector_date = str(sector_row._mapping["trade_date"]) if sector_row else "" latest_tracking_date = str(tracking_row._mapping["track_date"]) if tracking_row else "" return { "scan_running": _scan_running, "scan_mode": "intraday" if is_trading_hours() else "post_market", "is_trading": is_trading_hours(), "data_freshness": { "market_trade_date": latest_market_date, "sector_trade_date": latest_sector_date, "tracking_trade_date": latest_tracking_date, "last_recommendation_created_at": _fmt_dt(rec_row._mapping["created_at"]) if rec_row else "", "last_tracking_created_at": _fmt_dt(tracking_row._mapping["created_at"]) if tracking_row else "", "last_market_created_at": _fmt_dt(market_row._mapping["created_at"]) if market_row else "", "last_sector_created_at": _fmt_dt(sector_row._mapping["created_at"]) if sector_row else "", "status": "fresh" if latest_market_date else "empty", "message": ( f"最新市场日期 {latest_market_date}，最近跟踪 {latest_tracking_date or '暂无'}" if latest_market_date else "暂无市场缓存数据，请由管理员触发扫描。" ), "generated_at": datetime.now().isoformat(), }, "actions": [ {"key": "refresh", "label": "立即扫描", "admin_only": True}, {"key": "update_tracking", "label": "更新跟踪", "admin_only": True}, {"key": "generate_strategy_board", "label": "生成策略板", "admin_only": True}, {"key": "generate_strategy_iteration", "label": "生成策略复盘", "admin_only": True}, ], } @router.post("/generate-strategy-board") async def generate_strategy_board(_admin: dict = Depends(get_current_admin)): """管理员手动生成带 LLM 说明的策略看板""" from app.llm.strategy_board import build_strategy_board return await build_strategy_board(include_llm=True) @router.post("/generate-strategy-iteration") async def generate_strategy_iteration(limit: int = 50, _admin: dict = Depends(get_current_admin)): """管理员手动生成带 LLM 分析的策略复盘""" from app.llm.strategy_iteration import build_strategy_iteration_report return await build_strategy_iteration_report(limit=limit, include_llm=True) async def _overview_realtime(): """盘中：腾讯实时指数行情""" index_data = await tencent_client.get_index_realtime() result = [] name_map = { "000001.SH": "上证指数", "399001.SZ": "深证成指", "399006.SZ": "创业板指", } for code in ["000001.SH", "399001.SZ", "399006.SZ"]: data = index_data.get(code) if not data: continue result.append({ "name": name_map.get(code, data.get("name", code)), "code": code, "close": round(data["price"], 2), "pct_chg": round(data["pct_chg"], 2), "volume": round(data["volume"], 2), "realtime": True, }) return result def _overview_daily(): """盘后：Tushare 日线数据""" indices = { "上证指数": "000001.SH", "深证成指": "399001.SZ", "创业板指": "399006.SZ", } result = [] for name, code in indices.items(): df = tushare_client.get_index_daily(code, days=5) if df.empty: continue df = df.sort_values("trade_date") latest = df.iloc[-1] prev = df.iloc[-2] if len(df) > 1 else latest pct = (latest["close"] - prev["close"]) / prev["close"] * 100 result.append({ "name": name, "code": code, "close": round(float(latest["close"]), 2), "pct_chg": round(pct, 2), "volume": round(float(latest["vol"]), 2), "realtime": False, }) return result