ls증권 히스토리 구독 넣음

kis_trader/backtest/momentum_universe_timeline.py

@@ -60,6 +60,7 @@ def build_momentum_universe_timeline(
     debounce_sec: int = 30,
     strict: bool = False,
     strict_lag_minutes: int = 1,
+    history_source: str = "kiwoom",
 ) -> Optional[UniverseTimeline]:
     """모멘텀 유니버스 타임라인 — 전략 공통 ``build_universe_timeline`` 사용."""
     return build_universe_timeline(
@@ -69,6 +70,7 @@ def build_momentum_universe_timeline(
         debounce_sec=debounce_sec,
         strict=strict,
         strict_lag_minutes=strict_lag_minutes,
+        history_source=history_source,
     )
 
 
@@ -79,12 +81,26 @@ def attach_momentum_universe_timeline_to_params(
     end_ymd: str,
     strategy_id: str = MOMENTUM_STRATEGY_ID,
     use_saved_history: bool = True,
+    history_source: Optional[str] = None,
 ) -> Optional[UniverseTimeline]:
     """백테 params에 ``_momentum_universe_timeline`` 부착 (스캔 시각 유니버스)."""
     if not use_saved_history:
         return None
     if not momentum_backtest_universe_scan_at_enabled(params):
         return None
+    from kis_trader.backtest.universe_history_source import (
+        resolve_backtest_universe_history_source,
+    )
+
+    hs = resolve_backtest_universe_history_source(
+        history_source
+        if history_source is not None and str(history_source).strip() != ""
+        else (
+            params.get("_universe_history_source")
+            or params.get("universe_history_source")
+        )
+    )
+    params["_universe_history_source"] = hs
     debounce_sec = momentum_universe_exit_debounce_sec(params)
     # scan_at = 실매 get_universe_at(초) — 분봉 strict lag 는 minute 슬롯 전용
     timeline = build_momentum_universe_timeline(
@@ -94,6 +110,7 @@ def attach_momentum_universe_timeline_to_params(
         debounce_sec=debounce_sec,
         strict=False,
         strict_lag_minutes=0,
+        history_source=hs,
     )
     if timeline is not None:
         params["_momentum_universe_timeline"] = timeline
@@ -103,5 +120,6 @@ def attach_momentum_universe_timeline_to_params(
             "strict_lag_minutes": 0,
             "snapshots": timeline.snapshot_count,
             "mode": "scan_at",
+            "history_source": hs,
         }
     return timeline