stock-tracker

backend/app/routers/stocks.py

@@ -0,0 +1,24 @@
+from fastapi import APIRouter, Query
+from ..services.stock_data import search_stocks, get_kline, get_realtime_quote
+
+router = APIRouter()
+
+
+@router.get("/search")
+def search(q: str = Query(""), limit: int = Query(20)):
+    results = search_stocks(q, limit)
+    return {"data": results}
+
+
+@router.get("/kline/{code}")
+def kline(code: str, months: int = Query(12, ge=1, le=60)):
+    data = get_kline(code, months)
+    return {"data": data}
+
+
+@router.get("/realtime/{code}")
+def realtime(code: str):
+    quote = get_realtime_quote(code)
+    if quote is None:
+        return {"data": None, "error": "未找到该股票"}
+    return {"data": quote}