stock 실계좌 예수금 정보 추가

stock-lab/app/main.py

@@ -12,6 +12,7 @@ from .db import (
     init_db, save_articles, get_latest_articles,
     add_portfolio_item, get_all_portfolio, get_portfolio_item,
     update_portfolio_item, delete_portfolio_item,
+    upsert_broker_cash, get_all_broker_cash, get_broker_cash, delete_broker_cash,
 )
 from .scraper import fetch_market_news, fetch_major_indices, fetch_overseas_news
 from .price_fetcher import get_current_prices
@@ -155,10 +156,24 @@ class PortfolioUpdateRequest(BaseModel):
 
 @app.get("/api/portfolio")
 def get_portfolio():
-    """전체 포트폴리오 조회 (현재가 + 손익 포함)"""
+    """전체 포트폴리오 조회 (현재가 + 손익 + 예수금 포함)"""
     items = get_all_portfolio()
+    cash_rows = get_all_broker_cash()
+    total_cash = sum(r["cash"] for r in cash_rows)
+
     if not items:
-        return {"holdings": [], "summary": {"total_buy": 0, "total_eval": 0, "total_profit": 0, "total_profit_rate": 0.0}}
+        return {
+            "holdings": [],
+            "cash": cash_rows,
+            "summary": {
+                "total_buy": 0,
+                "total_eval": 0,
+                "total_profit": 0,
+                "total_profit_rate": 0.0,
+                "total_cash": total_cash,
+                "total_assets": total_cash,
+            },
+        }
 
     tickers = list({item["ticker"] for item in items})
     prices = get_current_prices(tickers)
@@ -196,11 +211,14 @@ def get_portfolio():
 
     return {
         "holdings": holdings,
+        "cash": cash_rows,
         "summary": {
             "total_buy": total_buy,
             "total_eval": total_eval,
             "total_profit": total_profit,
             "total_profit_rate": total_profit_rate,
+            "total_cash": total_cash,
+            "total_assets": total_eval + total_cash,
         },
     }
 
@@ -229,3 +247,31 @@ def delete_portfolio(item_id: int):
     return {"ok": True}
 
 
+# --- Broker Cash API ---
+
+class BrokerCashRequest(BaseModel):
+    broker: str
+    cash: int
+
+
+@app.get("/api/portfolio/cash")
+def list_broker_cash():
+    """증권사별 예수금 전체 조회"""
+    return get_all_broker_cash()
+
+
+@app.put("/api/portfolio/cash")
+def set_broker_cash(req: BrokerCashRequest):
+    """증권사 예수금 등록 또는 수정 (upsert)"""
+    upsert_broker_cash(req.broker, req.cash)
+    return {"ok": True, "broker": req.broker, "cash": req.cash}
+
+
+@app.delete("/api/portfolio/cash/{broker}")
+def remove_broker_cash(broker: str):
+    """증권사 예수금 삭제"""
+    if not delete_broker_cash(broker):
+        return JSONResponse(status_code=404, content={"error": "Broker not found"})
+    return {"ok": True}
+
+