feat: paginate KIS balance inquiries

2026-08-10 00:11:04 +09:00
parent 701fb06362
commit 0fa8f39be9
6 changed files with 297 additions and 37 deletions
@@ -652,41 +652,75 @@ class KISBroker:
            # Token issuance/header construction is distinct from a rejected
            # balance endpoint response and must never look like an empty book.
            raise BrokerBalanceError("token_issuance_failed") from None
-        try:
-            resp = httpx.get(
-                f"{self._base_url}/uapi/domestic-stock/v1/trading/inquire-balance",
-                headers=headers,
-                params=params,
-                timeout=self._timeout,
+
+        quantities: dict[str, int] = {}
+        seen_continuations: set[tuple[str, str]] = set()
+        continuation: Optional[tuple[str, str]] = None
+        while True:
+            request_headers = headers
+            if continuation is not None:
+                params["CTX_AREA_FK100"], params["CTX_AREA_NK100"] = continuation
+                request_headers = {**headers, "tr_cont": "N"}
+            try:
+                resp = httpx.get(
+                    f"{self._base_url}/uapi/domestic-stock/v1/trading/inquire-balance",
+                    headers=request_headers,
+                    params=params,
+                    timeout=self._timeout,
+                )
+                resp.raise_for_status()
+                payload = resp.json()
+            except httpx.HTTPStatusError as exc:
+                status = exc.response.status_code if exc.response is not None else None
+                raise BrokerBalanceError(_balance_http_error_code(status)) from None
+            except httpx.HTTPError:
+                raise BrokerBalanceError("transport_failed") from None
+            except (ValueError, TypeError):
+                raise BrokerBalanceError("malformed_response") from None
+
+            if not isinstance(payload, dict):
+                raise BrokerBalanceError("malformed_response")
+            if str(payload.get("rt_cd")) != "0":
+                raise BrokerBalanceError("broker_rejected")
+            rows = payload.get("output1")
+            if not isinstance(rows, list):
+                raise BrokerBalanceError("malformed_response")
+            for row in rows:
+                if not isinstance(row, dict):
+                    raise BrokerBalanceError("malformed_response")
+                qty = _safe_int(row.get("hldg_qty"))
+                if qty is None:
+                    raise BrokerBalanceError("malformed_response")
+                if qty <= 0:
+                    continue
+                symbol = str(row.get("pdno") or "").strip()
+                if not symbol:
+                    raise BrokerBalanceError("malformed_response")
+                quantities[symbol] = quantities.get(symbol, 0) + qty
+
+            response_headers = getattr(resp, "headers", None)
+            if response_headers is None:
+                raise BrokerBalanceError("malformed_response")
+            tr_cont = response_headers.get("tr_cont")
+            if not isinstance(tr_cont, str):
+                raise BrokerBalanceError("malformed_response")
+            tr_cont = tr_cont.strip().upper()
+            if tr_cont in {"D", "E"}:
+                return [
+                    BalanceItem(symbol=symbol, hldg_qty=qty)
+                    for symbol, qty in quantities.items()
+                ]
+            if tr_cont not in {"F", "M"}:
+                raise BrokerBalanceError("malformed_response")
+
+            next_continuation = (
+                str(payload.get("ctx_area_fk100") or "").strip(),
+                str(payload.get("ctx_area_nk100") or "").strip(),
            )
-            resp.raise_for_status()
-            payload = resp.json()
-        except httpx.HTTPStatusError as exc:
-            status = exc.response.status_code if exc.response is not None else None
-            raise BrokerBalanceError(_balance_http_error_code(status)) from None
-        except httpx.HTTPError:
-            raise BrokerBalanceError("transport_failed") from None
-        except (ValueError, TypeError):
-            raise BrokerBalanceError("malformed_response") from None
-        if not isinstance(payload, dict):
-            raise BrokerBalanceError("malformed_response")
-        if str(payload.get("rt_cd")) != "0":
-            raise BrokerBalanceError("broker_rejected")
-        rows = payload.get("output1")
-        if not isinstance(rows, list):
-            raise BrokerBalanceError("malformed_response")
-        out: list[BalanceItem] = []
-        for row in rows:
-            if not isinstance(row, dict):
+            if not all(next_continuation) or next_continuation in seen_continuations:
                raise BrokerBalanceError("malformed_response")
-            symbol = str(row.get("pdno") or "").strip()
-            qty = _safe_int(row.get("hldg_qty"))
-            if qty is None:
-                raise BrokerBalanceError("malformed_response")
-            if not symbol or qty <= 0:
-                continue
-            out.append(BalanceItem(symbol=symbol, hldg_qty=qty))
-        return out
+            seen_continuations.add(next_continuation)
+            continuation = next_continuation

    def get_available_cash(self, *, price_check_code: str = "005930") -> int:
        """Return KIS cash-only buy capacity (`nrcvb_buy_amt`).