merge(feat/kis-auto-fetch): feat/kis-auto-fetch -> main

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Author: illenne77

pages/2_포트폴리오_대시보드.py

@@ -6,6 +6,7 @@
 
 from __future__ import annotations
 
+import os
 from decimal import Decimal
 
 import pandas as pd
@@ -28,23 +29,97 @@
         help="올해 이미 실현한 양도차익. 0이면 기본공제 250만원 전액 미사용.",
     )
 
+# ── KIS 계좌 자동 조회 (로컬 전용) ───────────────────────────
+with st.expander("🔗 KIS 증권계좌 자동 조회 (로컬 실행 전용)", expanded=False):
+    st.caption(
+        "KIS OpenAPI를 통해 보유 종목을 자동으로 불러옵니다.\n\n"
+        "⚠️ Streamlit Cloud에서는 KIS API 접근이 제한됩니다. 로컬에서만 동작합니다.\n\n"
+        "사전 준비: `.env` 파일에 KIS 키 설정 후 `python scripts/kis_token.py live` 실행 필요."
+    )
+
+    _kis_key = os.environ.get("KIS_APP_KEY") or os.environ.get("KIS_LIVE_APP_KEY")
+    _kis_secret = os.environ.get("KIS_APP_SECRET") or os.environ.get("KIS_LIVE_APP_SECRET")
+    _kis_account = os.environ.get("KIS_ACCOUNT")
+    _kis_ready = bool(_kis_key and _kis_secret and _kis_account)
+
+    if not _kis_ready:
+        st.warning("KIS_APP_KEY, KIS_APP_SECRET, KIS_ACCOUNT 환경변수를 설정해야 합니다.")
+    else:
+        from sentinelq.adapters.kis_history import SECRETS_DIR
+
+        _token_ok = (SECRETS_DIR / "kis_token_live.json").exists()
+        if not _token_ok:
+            st.warning(
+                "KIS 토큰이 없습니다. 먼저 터미널에서 실행하세요:\n"
+                "```\npython scripts/kis_token.py live\n```"
+            )
+
+        if st.button("KIS 잔고 조회 🔄", disabled=not _token_ok):
+            if not os.environ.get("KIS_LIVE_APP_KEY"):
+                os.environ["KIS_LIVE_APP_KEY"] = _kis_key
+            if not os.environ.get("KIS_LIVE_APP_SECRET"):
+                os.environ["KIS_LIVE_APP_SECRET"] = _kis_secret
+            if not os.environ.get("KIS_LIVE_BASE_URL"):
+                os.environ["KIS_LIVE_BASE_URL"] = "https://openapi.koreainvestment.com:9443"
+            os.environ["SENTINELQ_LIVE_ALLOW"] = "1"
+
+            from sentinelq.adapters.kis_history import fetch_balance
+
+            with st.spinner("KIS 잔고 조회 중..."):
+                try:
+                    _auto_holdings = fetch_balance(env="live", confirm_live=True)
+                except Exception as exc:
+                    st.error(f"조회 오류: {exc}")
+                    _err = str(exc).lower()
+                    if any(k in _err for k in ("timed out", "network", "urlopen", "connection")):
+                        st.info(
+                            "💡 Streamlit Cloud에서는 KIS API 접근이 제한됩니다. "
+                            "로컬에서 실행하세요:\n"
+                            "```bash\nstreamlit run streamlit_app.py\n```"
+                        )
+                    st.stop()
+
+            if not _auto_holdings:
+                st.info("조회된 잔고가 없습니다.")
+            else:
+                st.success(f"{len(_auto_holdings)}개 종목 조회 완료. 아래 표에 자동 입력됩니다.")
+                st.session_state["kis_holdings"] = _auto_holdings
+                st.rerun()
+
+# ── 보유 종목 입력 테이블 ─────────────────────────────────────
 st.subheader("📋 보유 종목 입력")
-st.caption("아래 표에 직접 입력하거나 편집하세요.")
-
-# ── 기본 예시 데이터 ─────────────────────────────────────────
-_DEFAULT = pd.DataFrame(
-    {
-        "종목코드": ["005930", "AAPL"],
-        "종목명": ["삼성전자", "Apple Inc"],
-        "시장": ["KR", "US"],
-        "수량": [10, 5],
-        "평균단가(원)": [70_000, 1_800_000],
-        "현재가(원)": [78_000, 2_100_000],
-    }
-)
+
+if "kis_holdings" in st.session_state:
+    st.caption("KIS 자동 조회 데이터가 로드됐습니다. 수정 후 계산하기를 누르세요.")
+    _h = st.session_state["kis_holdings"]
+    _initial = pd.DataFrame(
+        [
+            {
+                "종목코드": h.ticker,
+                "종목명": h.name,
+                "시장": h.market,
+                "수량": h.quantity,
+                "평균단가(원)": int(h.avg_price_krw),
+                "현재가(원)": int(h.current_price_krw),
+            }
+            for h in _h
+        ]
+    )
+else:
+    st.caption("아래 표에 직접 입력하거나 편집하세요.")
+    _initial = pd.DataFrame(
+        {
+            "종목코드": ["005930", "AAPL"],
+            "종목명": ["삼성전자", "Apple Inc"],
+            "시장": ["KR", "US"],
+            "수량": [10, 5],
+            "평균단가(원)": [70_000, 1_800_000],
+            "현재가(원)": [78_000, 2_100_000],
+        }
+    )
 
 edited = st.data_editor(
-    _DEFAULT,
+    _initial,
     num_rows="dynamic",
     use_container_width=True,
     column_config={
@@ -60,7 +135,6 @@
         st.warning("종목을 1개 이상 입력해 주세요.")
         st.stop()
 
-    # HoldingRecord 생성
     from sentinelq.adapters.kis_history import HoldingRecord
     from sentinelq.portfolio.after_tax import calculate_after_tax
 
@@ -134,7 +208,6 @@
     else:
         st.info("보유 종목 없음")
 
-    # 세션 상태에 포트폴리오 저장 (리밸런싱 페이지 연동)
     st.session_state["portfolio"] = portfolio
     st.caption("※ 이 포트폴리오 데이터가 리밸런싱 계산기 페이지에서 자동 로드됩니다.")
 

sentinelq/adapters/kis_history.py

@@ -659,12 +659,26 @@ def inquire_overseas_balance(
     return results
 
 
+def fetch_balance(
+    *,
+    env: Env = "live",
+    account: str | None = None,
+    confirm_live: bool = False,
+) -> list[HoldingRecord]:
+    """국내·해외 잔고 통합 조회 (포트폴리오 대시보드용)."""
+    kr = inquire_domestic_balance(env=env, account=account, confirm_live=confirm_live)
+    us = inquire_overseas_balance(env=env, account=account, confirm_live=confirm_live)
+    return kr + us
+
+
 __all__ = [
+    "SECRETS_DIR",
     "Env",
     "HoldingRecord",
     "KisApiError",
     "ProfitRecord",
     "Transaction",
+    "fetch_balance",
     "inquire_domestic_balance",
     "inquire_domestic_daily_trans",
     "inquire_overseas_balance",

tests/test_nts_form.py

@@ -238,15 +238,15 @@ def test_summary_csv_has_header_and_key_fields_ac9():
     csv_str = export_summary_csv(form)
 
     lines = csv_str.strip().splitlines()
-    assert lines[0] == "field,value"
+    assert lines[0] == "항목,금액(원)"
 
     fields = {row.split(",")[0] for row in lines[1:]}
     for key in (
-        "tax_year",
-        "total_proceeds_krw",
-        "national_tax_krw",
-        "local_tax_krw",
-        "total_tax_krw",
+        "과세연도",
+        "양도가액",
+        "국세(양도소득세)",
+        "지방소득세",
+        "납부세액 합계",
     ):
         assert key in fields, f"summary CSV missing field: {key}"
 
@@ -263,13 +263,13 @@ def test_detail_csv_per_line_columns_ac10():
     lines = csv_str.strip().splitlines()
     header_cols = lines[0].split(",")
     for col in (
-        "market",
-        "ticker",
-        "sell_date",
-        "quantity",
-        "proceeds_krw",
-        "acquisition_cost_krw",
-        "realized_gain_krw",
+        "시장",
+        "종목코드",
+        "매도일",
+        "수량",
+        "양도가액(원)",
+        "취득가액(원)",
+        "양도차익(원)",
     ):
         assert col in header_cols