ls증권 히스토리 구독 넣음

2026-07-30 18:05:07 +09:00
parent 61bec4bd1d
commit 67eab24603
1593 changed files with 135733 additions and 1232 deletions
--- a/kis_trader/utils/kis_ws_session_windows.py
+++ b/kis_trader/utils/kis_ws_session_windows.py
@@ -0,0 +1,149 @@
+"""
+KIS WebSocket approval 세션 시간 분할 (국내 ↔ 해외)
+====================================================
+KIS 정책: approval_key : WebSocket 세션 = 1 : 1
+국내 H0STCNT0 / 해외 HDFSCNT0 는 동일 실전 URL(ops:21000)을 쓰므로
+**동시에 두 소켓을 열면** 뒤쪽(보통 해외)이 즉시 끊긴다.
+
+운용:
+  - 국내 hold 창: 국내 WS만 연결 (기본 07:00~20:00 — 장 전후·틱 적재 여유)
+  - 해외 hold 창: 해외 WS만 연결 (기본 21:00~06:00 — 미국장 전후 틱 적재 여유)
+  - 그 사이 갭(20:00~21:00, 06:00~07:00): 양쪽 모두 close (세션 비움·핸드오버)
+  - approval REST 재발급: 기존 6h 금지 · 24h 유효 · 응급 하드캡 유지
+    (이 모듈은 재발급하지 않음 — 소켓 close/open 만)
+
+env (DB 등록):
+  KIS_WS_KR_HOLD_START_HM / KIS_WS_KR_HOLD_END_HM   (기본 700 / 2000)
+  KIS_WS_US_HOLD_START_HM / KIS_WS_US_HOLD_END_HM   (기본 2100 / 600)
+  KIS_WS_SESSION_GUARD_SEC  — 연결 중 창 이탈 시 close 주기
+"""
+from __future__ import annotations
+
+import datetime as _dt
+from typing import Optional, Tuple
+
+from .env import get_env_float, get_env_int
+from .session_hm import hm_in_trading_window
+
+
+def kr_ws_hold_bounds() -> Tuple[int, int]:
+    """국내 WS 세션 점유 HHMM (당일 구간). 기본 0700~2000."""
+    start = int(get_env_int("KIS_WS_KR_HOLD_START_HM", 700) or 700)
+    end = int(get_env_int("KIS_WS_KR_HOLD_END_HM", 2000) or 2000)
+    if start <= 0:
+        start = 700
+    if end <= 0:
+        end = 2000
+    return start, end
+
+
+def us_ws_hold_bounds() -> Tuple[int, int]:
+    """해외 WS 세션 점유 HHMM (자정 넘김). 기본 2100~0600."""
+    start = int(get_env_int("KIS_WS_US_HOLD_START_HM", 2100) or 2100)
+    end = int(get_env_int("KIS_WS_US_HOLD_END_HM", 600) or 600)
+    if start <= 0:
+        start = 2100
+    if end <= 0:
+        end = 600
+    return start, end
+
+
+def session_guard_interval_sec() -> float:
+    """연결 유지 중 hold 창 이탈 감시 주기(초)."""
+    return max(5.0, float(get_env_float("KIS_WS_SESSION_GUARD_SEC", 15.0) or 15.0))
+
+
+def _hm_now(now: Optional[_dt.datetime] = None) -> int:
+    n = now or _dt.datetime.now()
+    return int(n.hour * 100 + n.minute)
+
+
+def in_kr_ws_hold_window(now: Optional[_dt.datetime] = None) -> bool:
+    """True = 국내 WS 가 approval 세션을 점유해도 되는 시간 (월~금)."""
+    n = now or _dt.datetime.now()
+    if n.weekday() >= 5:
+        return False
+    start, end = kr_ws_hold_bounds()
+    return hm_in_trading_window(_hm_now(n), start, end, wrap_midnight=False)
+
+
+def in_us_ws_hold_window(now: Optional[_dt.datetime] = None) -> bool:
+    """
+    True = 해외 WS 가 approval 세션을 점유해도 되는 시간.
+    - start~23:59: 월~금
+    - 00:00~end: 화~토 (미국장 새벽 마감)
+    """
+    n = now or _dt.datetime.now()
+    wd = n.weekday()
+    hm = _hm_now(n)
+    start, end = us_ws_hold_bounds()
+    # 자정 넘김: start > end
+    if start > end:
+        if hm >= start:
+            return 0 <= wd <= 4
+        if hm < end:
+            return 1 <= wd <= 5
+        return False
+    # 비정상(같은날 구간) — 월~금만
+    if n.weekday() >= 5:
+        return False
+    return hm_in_trading_window(hm, start, end, wrap_midnight=False)
+
+
+def seconds_until_kr_ws_open(now: Optional[_dt.datetime] = None) -> float:
+    """다음 국내 hold 시작까지 초 (최소 60). hold 중이면 짧은 쿨다운."""
+    n = now or _dt.datetime.now()
+    if in_kr_ws_hold_window(n):
+        return max(30.0, float(get_env_float("KIS_WS_INSTANT_DROP_COOLDOWN_SEC", 90.0) or 90.0))
+    start, _end = kr_ws_hold_bounds()
+    sh, sm = divmod(int(start), 100)
+    hm = _hm_now(n)
+    target = n.replace(hour=sh, minute=sm, second=0, microsecond=0)
+    if n.weekday() >= 5:
+        # 토→월(+2), 일→월(+1)
+        days = (0 - n.weekday()) % 7
+        if days == 0:
+            days = 7
+        target += _dt.timedelta(days=days)
+    elif hm >= start:
+        target += _dt.timedelta(days=1)
+        while target.weekday() >= 5:
+            target += _dt.timedelta(days=1)
+    return max(60.0, (target - n).total_seconds())
+
+
+def seconds_until_us_ws_open(now: Optional[_dt.datetime] = None) -> float:
+    """다음 해외 hold 시작까지 초 (최소 60). hold 중이면 짧은 쿨다운."""
+    n = now or _dt.datetime.now()
+    if in_us_ws_hold_window(n):
+        return max(30.0, float(get_env_float("KIS_WS_INSTANT_DROP_COOLDOWN_SEC", 90.0) or 90.0))
+    start, end = us_ws_hold_bounds()
+    sh, sm = divmod(int(start), 100)
+    hm = _hm_now(n)
+    target = n.replace(hour=sh, minute=sm, second=0, microsecond=0)
+    if start > end:
+        # 주간 갭(end~start): 오늘 밤 start (단, 토요일이면 월)
+        if end <= hm < start:
+            if n.weekday() >= 5:
+                days = (0 - n.weekday()) % 7
+                if days == 0:
+                    days = 7
+                target += _dt.timedelta(days=days)
+        elif hm >= start:
+            # 금요 밤 hold 가능 구간인데 hold False = 이상 → 다음 평일
+            # 토요 밤: 다음 월
+            target += _dt.timedelta(days=1)
+            while target.weekday() >= 5:
+                target += _dt.timedelta(days=1)
+        else:
+            # hm < end, hold 아님 (예: 일요일 새벽) → 오늘(일) 밤 스킵 → 월 start
+            if n.weekday() == 6:
+                target += _dt.timedelta(days=1)
+            while target.weekday() >= 5:
+                target += _dt.timedelta(days=1)
+    else:
+        if hm >= start or n.weekday() >= 5:
+            target += _dt.timedelta(days=1)
+        while target.weekday() >= 5:
+            target += _dt.timedelta(days=1)
+    return max(60.0, (target - n).total_seconds())