feat: 자산웹 실시간 시세·평가손익 (키움 WebSocket)
- realtime_hub.py: 키움 WS 허브(0B 체결가·1h VI), behive_web 데몬 스레드 + 메모리 상태 - /api/realtime/quotes 엔드포인트 (허브 메모리만, 키움 호출 0) - 보유행 현재가·평가손익·평가금액·등락률(NXT종가 기준)·비중 실시간 - KPI 총평가금액·총평가손익·순자산·당일평가손익 실시간 (owner 탭 합계→자산정보 탭 카드 연동) - fill_watcher 체결 시 fill_signal 기록 → 자산웹 패널 자동 새로고침(수량·예수금·당일정산) - 감시종목 탭 자동갱신 카운트다운 활성화 - 안전: kill-switch BEHIVE_REALTIME=0, 허브 콜드/실패 시 REST 폴백, 매매 경로 무관 Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
hyowons
@@ -0,0 +1,318 @@
|
||||
#!/usr/bin/env python3
|
||||
"""키움 실시간 시세 허브 (WebSocket) — A단계 단독 모듈.
|
||||
|
||||
조회 전용 REST(kiwoom_client)와 별개로, 키움 실시간 시세 WebSocket을 상시 청취해
|
||||
최신 현재가(0B 주식체결)·VI 발동/해제(1h)를 메모리에 들고 있다가 읽어주는 컴포넌트.
|
||||
|
||||
behive_web 프로세스 안의 데몬 스레드로 띄울 예정이지만(B단계), 이 파일 자체는
|
||||
behive_web을 전혀 import/수정하지 않는다. 단독 실행으로 실접속·파싱을 먼저 검증한다.
|
||||
|
||||
python3 realtime_hub.py test 005930 000660 # 지정 종목 실시간 수신을 콘솔에 출력
|
||||
|
||||
설계:
|
||||
- 연결 1개(시세는 계좌 무관 → 토큰 아무 계좌나 재사용)
|
||||
- 흐름: connect → LOGIN → REG(구독) → 수신 루프(PING 에코·REAL 파싱) → 끊기면 백오프 재연결+재구독
|
||||
- 메모리 상태 저장소(lock): quotes{code:{...}}, vi{code:{...}}
|
||||
- hub가 콜드/끊김이어도 호출측은 REST로 폴백 가능하도록, 읽기 메서드는 없으면 None 반환
|
||||
|
||||
키움 WS 프로토콜:
|
||||
- 엔드포인트: wss://api.kiwoom.com:10000/api/dostk/websocket (운영)
|
||||
- LOGIN/PING 핸드셰이크는 공식 PDF에 표가 없어 키움 공식 샘플 규약을 따름:
|
||||
LOGIN 송신: {"trnm":"LOGIN","token":"<access_token>"} (Bearer 접두어 없이 raw 토큰)
|
||||
서버가 {"trnm":"PING"} 주기 송신 → 같은 메시지 그대로 에코
|
||||
- REG/REAL 스펙·필드코드는 PDF(국내주식>실시간시세) 기준:
|
||||
0B 주식체결: 10=현재가 11=전일대비 12=등락율 20=체결시간 16/17/18=시고저 9081=거래소
|
||||
1h VI발동/해제: 9068=발동구분 1221=발동가격 1224=해제시각 1225=적용구분 1489=등락율 9069=방향
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import sys
|
||||
import threading
|
||||
import time
|
||||
from datetime import datetime, timezone, timedelta
|
||||
|
||||
import websocket # websocket-client (동기식). 기존 threading 서버와 궁합.
|
||||
|
||||
import kiwoom_client as kc
|
||||
|
||||
KST = timezone(timedelta(hours=9))
|
||||
|
||||
# 재연결 백오프 (초) — 점증, 상한 30s
|
||||
_BACKOFF_START = 2.0
|
||||
_BACKOFF_MAX = 30.0
|
||||
# 토큰 만료 전 선제 재연결 (초). 키움 토큰 ~30분 → 25분마다 새 연결로 갈아탐.
|
||||
_MAX_CONN_AGE = 25 * 60
|
||||
# recv 타임아웃 — 이 시간 내 아무것도 안 오면 루프 돌며 stop/age 체크
|
||||
_RECV_TIMEOUT = 5.0
|
||||
|
||||
|
||||
def _ws_url() -> str:
|
||||
"""REST base_url에서 WS 엔드포인트 도출. https→wss, :10000, /api/dostk/websocket."""
|
||||
base = kc.base_url() # 예: https://api.kiwoom.com 또는 https://mockapi.kiwoom.com
|
||||
host = base.split('://', 1)[-1].rstrip('/')
|
||||
return f'wss://{host}:10000/api/dostk/websocket'
|
||||
|
||||
|
||||
def _abs_int(s) -> int:
|
||||
"""부호 포함 문자열 → 절대값 int (현재가·호가용)."""
|
||||
return abs(kc._to_int(s))
|
||||
|
||||
|
||||
class RealtimeHub:
|
||||
"""키움 실시간 시세 허브. start()로 데몬 스레드 기동, get_quote/get_vi로 읽기."""
|
||||
|
||||
def __init__(self, account_label: str | None = None):
|
||||
self._label = account_label or kc._default_account_label()
|
||||
self._lock = threading.Lock()
|
||||
self._quotes: dict[str, dict] = {} # code -> {price, change, pct, ts, exchange}
|
||||
self._vi: dict[str, dict] = {} # code -> {active, kind, trigger_price, ...}
|
||||
self._subs: set[str] = set() # 현재 구독 종목코드(정규화)
|
||||
self._ws: websocket.WebSocket | None = None
|
||||
self._ws_lock = threading.Lock() # send 직렬화
|
||||
self._stop = threading.Event()
|
||||
self._thread: threading.Thread | None = None
|
||||
self._connected = threading.Event()
|
||||
self.last_error: str | None = None
|
||||
self.connected_at: float = 0.0
|
||||
|
||||
# ---- 공개 읽기 API (없으면 None → 호출측 REST 폴백) ----
|
||||
def get_quote(self, code: str) -> dict | None:
|
||||
with self._lock:
|
||||
q = self._quotes.get(kc._clean_code(code))
|
||||
return dict(q) if q else None
|
||||
|
||||
def get_quotes(self, codes: list[str]) -> dict[str, dict]:
|
||||
with self._lock:
|
||||
out = {}
|
||||
for c in codes:
|
||||
cc = kc._clean_code(c)
|
||||
if cc in self._quotes:
|
||||
out[cc] = dict(self._quotes[cc])
|
||||
return out
|
||||
|
||||
def get_vi(self, code: str) -> dict | None:
|
||||
with self._lock:
|
||||
v = self._vi.get(kc._clean_code(code))
|
||||
return dict(v) if v else None
|
||||
|
||||
def active_vis(self) -> dict[str, dict]:
|
||||
with self._lock:
|
||||
return {c: dict(v) for c, v in self._vi.items() if v.get('active')}
|
||||
|
||||
def is_connected(self) -> bool:
|
||||
return self._connected.is_set()
|
||||
|
||||
# ---- 구독 관리 ----
|
||||
def set_subscriptions(self, codes: list[str]) -> None:
|
||||
"""구독 종목 목록을 통째로 교체. 연결돼 있으면 즉시 REG/REMOVE 반영."""
|
||||
new = {kc._clean_code(c) for c in codes if c}
|
||||
with self._lock:
|
||||
add = new - self._subs
|
||||
rem = self._subs - new
|
||||
self._subs = new
|
||||
if self._connected.is_set():
|
||||
if add:
|
||||
self._send_reg(sorted(add))
|
||||
if rem:
|
||||
self._send_remove(sorted(rem))
|
||||
|
||||
# ---- 생명주기 ----
|
||||
def start(self) -> None:
|
||||
if self._thread and self._thread.is_alive():
|
||||
return
|
||||
self._stop.clear()
|
||||
self._thread = threading.Thread(target=self._run, name='realtime-hub', daemon=True)
|
||||
self._thread.start()
|
||||
|
||||
def stop(self) -> None:
|
||||
self._stop.set()
|
||||
with self._ws_lock:
|
||||
if self._ws:
|
||||
try:
|
||||
self._ws.close()
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
# ---- 내부: 연결 루프 ----
|
||||
def _run(self) -> None:
|
||||
backoff = _BACKOFF_START
|
||||
while not self._stop.is_set():
|
||||
try:
|
||||
self._connect_once()
|
||||
backoff = _BACKOFF_START # 정상 종료(선제 재연결 등) → 백오프 리셋
|
||||
except Exception as e:
|
||||
self.last_error = f'{type(e).__name__}: {e}'
|
||||
self._connected.clear()
|
||||
sys.stderr.write(f'[realtime-hub] 연결 끊김/실패: {self.last_error}\n')
|
||||
if self._stop.is_set():
|
||||
break
|
||||
time.sleep(backoff)
|
||||
backoff = min(backoff * 2, _BACKOFF_MAX)
|
||||
|
||||
def _connect_once(self) -> None:
|
||||
url = _ws_url()
|
||||
token = kc.issue_token(self._label) # 매 연결마다 fresh (캐시·만료는 issue_token이 관리)
|
||||
ws = websocket.create_connection(url, timeout=10)
|
||||
ws.settimeout(_RECV_TIMEOUT)
|
||||
with self._ws_lock:
|
||||
self._ws = ws
|
||||
try:
|
||||
# 1) LOGIN
|
||||
ws.send(json.dumps({'trnm': 'LOGIN', 'token': token}))
|
||||
login = json.loads(ws.recv())
|
||||
if login.get('trnm') == 'LOGIN' and login.get('return_code') not in (0, '0', None):
|
||||
raise RuntimeError(f'LOGIN 실패: {login.get("return_msg")} ({login})')
|
||||
self._connected.set()
|
||||
self.connected_at = time.time()
|
||||
self.last_error = None
|
||||
sys.stderr.write(f'[realtime-hub] LOGIN 성공 @ {url}\n')
|
||||
|
||||
# 2) 현재 구독 종목 REG
|
||||
with self._lock:
|
||||
codes = sorted(self._subs)
|
||||
if codes:
|
||||
self._send_reg(codes)
|
||||
|
||||
# 3) 수신 루프
|
||||
while not self._stop.is_set():
|
||||
if time.time() - self.connected_at > _MAX_CONN_AGE:
|
||||
sys.stderr.write('[realtime-hub] 토큰 만료 전 선제 재연결\n')
|
||||
return # 정상 종료 → _run이 즉시 재연결
|
||||
try:
|
||||
raw = ws.recv()
|
||||
except websocket.WebSocketTimeoutException:
|
||||
continue
|
||||
if not raw:
|
||||
continue
|
||||
self._handle(raw)
|
||||
finally:
|
||||
self._connected.clear()
|
||||
with self._ws_lock:
|
||||
self._ws = None
|
||||
try:
|
||||
ws.close()
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
# ---- 송신 ----
|
||||
def _send(self, payload: dict) -> None:
|
||||
with self._ws_lock:
|
||||
if self._ws is None:
|
||||
return
|
||||
self._ws.send(json.dumps(payload))
|
||||
|
||||
def _send_reg(self, codes: list[str]) -> None:
|
||||
# 0B(체결) + 1h(VI) 동시 등록. refresh=1 → 기존 구독 유지하며 추가.
|
||||
self._send({
|
||||
'trnm': 'REG', 'grp_no': '1', 'refresh': '1',
|
||||
'data': [{'item': codes, 'type': ['0B', '1h']}],
|
||||
})
|
||||
|
||||
def _send_remove(self, codes: list[str]) -> None:
|
||||
self._send({
|
||||
'trnm': 'REMOVE', 'grp_no': '1',
|
||||
'data': [{'item': codes, 'type': ['0B', '1h']}],
|
||||
})
|
||||
|
||||
# ---- 수신 처리 ----
|
||||
def _handle(self, raw: str) -> None:
|
||||
try:
|
||||
msg = json.loads(raw)
|
||||
except (ValueError, TypeError):
|
||||
return
|
||||
trnm = msg.get('trnm')
|
||||
if trnm == 'PING':
|
||||
# PING은 받은 그대로 에코 (키움 keepalive 규약)
|
||||
with self._ws_lock:
|
||||
if self._ws is not None:
|
||||
self._ws.send(raw)
|
||||
return
|
||||
if trnm == 'REAL':
|
||||
for blk in msg.get('data') or []:
|
||||
self._handle_real(blk)
|
||||
return
|
||||
if trnm in ('REG', 'REMOVE'):
|
||||
rc = msg.get('return_code')
|
||||
if rc not in (0, '0', None):
|
||||
sys.stderr.write(f'[realtime-hub] {trnm} 거부: rc={rc} msg={msg.get("return_msg")}\n')
|
||||
else:
|
||||
sys.stderr.write(f'[realtime-hub] {trnm} ok (rc={rc})\n')
|
||||
return
|
||||
# LOGIN 응답 등 기타는 무시
|
||||
|
||||
def _handle_real(self, blk: dict) -> None:
|
||||
typ = blk.get('type')
|
||||
code = kc._clean_code(blk.get('item') or '')
|
||||
vals = blk.get('values') or {}
|
||||
if not code:
|
||||
return
|
||||
if typ == '0B': # 주식체결 → 현재가
|
||||
with self._lock:
|
||||
self._quotes[code] = {
|
||||
'price': _abs_int(vals.get('10')),
|
||||
'change': kc._to_int(vals.get('11')),
|
||||
'pct': kc._to_float(vals.get('12')),
|
||||
'exchange': (vals.get('9081') or '').strip(),
|
||||
'ts': datetime.now(KST).isoformat(),
|
||||
}
|
||||
elif typ == '1h': # VI 발동/해제
|
||||
# active 판정: 9068 VI발동구분. 해제시각(1224) 유무와 함께 실증 확인 예정(A단계).
|
||||
kind_raw = (vals.get('9068') or '').strip()
|
||||
with self._lock:
|
||||
self._vi[code] = {
|
||||
'active': kind_raw not in ('', '0', '2'), # ⚠️ 매핑 실증 확인 필요
|
||||
'vi_gubun': kind_raw,
|
||||
'apply_kind': (vals.get('1225') or '').strip(), # 정적/동적/동적+정적
|
||||
'trigger_price': _abs_int(vals.get('1221')),
|
||||
'trigger_pct': kc._to_float(vals.get('1489')),
|
||||
'release_time': (vals.get('1224') or '').strip(),
|
||||
'direction': (vals.get('9069') or '').strip(),
|
||||
'ts': datetime.now(KST).isoformat(),
|
||||
}
|
||||
|
||||
|
||||
# ---------------- CLI (A단계 단독 검증) ----------------
|
||||
def _cmd_test(codes: list[str]) -> int:
|
||||
if not codes:
|
||||
codes = ['005930', '000660'] # 삼성전자, SK하이닉스
|
||||
print(f'[test] WS URL = {_ws_url()}')
|
||||
print(f'[test] 구독 종목 = {codes} (Ctrl-C 종료)')
|
||||
hub = RealtimeHub()
|
||||
hub.set_subscriptions(codes)
|
||||
hub.start()
|
||||
last_print: dict[str, str] = {}
|
||||
try:
|
||||
while True:
|
||||
time.sleep(1)
|
||||
if not hub.is_connected():
|
||||
continue
|
||||
for c in codes:
|
||||
cc = kc._clean_code(c)
|
||||
q = hub.get_quote(cc)
|
||||
if q:
|
||||
line = f"{q['price']:,} ({q['pct']:+.2f}%) {q['exchange']}"
|
||||
if last_print.get(cc) != line:
|
||||
print(f' [{cc}] {line} @ {q["ts"][11:19]}')
|
||||
last_print[cc] = line
|
||||
v = hub.get_vi(cc)
|
||||
if v and v.get('active'):
|
||||
vk = f"VI:{cc} {v['apply_kind']} 발동가 {v['trigger_price']:,} ({v['trigger_pct']:+.2f}%)"
|
||||
if last_print.get(f'vi_{cc}') != vk:
|
||||
print(f' ⚡ {vk}')
|
||||
last_print[f'vi_{cc}'] = vk
|
||||
except KeyboardInterrupt:
|
||||
print('\n[test] 종료')
|
||||
hub.stop()
|
||||
return 0
|
||||
|
||||
|
||||
def main(argv: list[str]) -> int:
|
||||
if not argv or argv[0] != 'test':
|
||||
print('usage: python3 realtime_hub.py test [code ...]', file=sys.stderr)
|
||||
return 2
|
||||
return _cmd_test(argv[1:])
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
sys.exit(main(sys.argv[1:]))
|
||||
Reference in New Issue
Block a user