hoon/kis_bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 새로운 안전 규칙 및 최적화 적용을 통한 트레이딩 시스템 개선

Browse Source 
변경 사항 (Changes):

구문 오류(Syntax error) 및 토큰 낭비를 방지하기 위해 에이전트 쉘(Agent shell)과 파이썬 코드 스니펫에 다수의 신규 안전 규칙(Safety rules)을 추가함.

스키마 검증 및 적절한 SQL 포맷팅을 보장하기 위해 임시(Ad-hoc) 데이터베이스 쿼리 작성 가이드라인을 도입함.

코드 수정 후 UI 기능이 정상 작동하는지 확인하기 위해, 백테스트 웹 서비스 재시작 및 브라우저 검증에 대한 새로운 규칙을 구현함.

시스템 전반의 무결성(Integrity)을 유지하기 위해 실전 매매(Live trading), 웹 백테스팅, 파라미터 탐색(Parameter searches) 간의 일관성 검사(Consistency checks) 체계를 확립함.

기대 효과 (Impact):

이러한 개선 사항들은 트레이딩 시스템의 견고성(Robustness)과 신뢰성을 향상시키며, 에러 발생을 최소화하고 다양한 시스템 컴포넌트 간의 원활한 상호작용을 보장함.
This commit is contained in:
Your Name
2026-07-17 01:09:09 +09:00
parent a4626e0351
commit fc27e726f9
151 changed files with 20718 additions and 6450 deletions
Download Patch File Download Diff File Expand all files Collapse all files

215
kis_trader/network/kiwoom_condition_manager.py
View File

@@ -8,9 +8,14 @@ kis_trader/network/kiwoom_condition_manager.py — 키움 조건검색 기반
- CNSRLST : 서버 저장 조건식 목록 (name → seq 해결)
- CNSRREQ : 실시간 조건검색 등록 (search_type="1") → 초기 매칭 + 이후 REAL push
- REAL : 편입(843="I") / 이탈(843="D"), 종목코드는 9001
- CNSRCLR : 실시간 해제
- CNSRCLR : 실시간 해제 (재LOGIN/재등록 전 선행 — Bye 후 유령등록·900003 방지)
(이 경로는 ``_test_kiwoom_condition_realtime.py`` 에서 실계정으로 검증 완료)
재접속 복구:
* 키움 WS ``Bye`` 후에도 계정에 조건검색이 남을 수 있음 → ``KIWOOM_CNSRCLR_BEFORE_REQ``
(기본 true) 로 CNSRREQ 전에 CNSRCLR.
* rc=900003 은 seq 당 1회만 CLR→REQ (무한 재시도 금지).
설계 (최소 침습):
* ``ConditionSearchManager`` 를 **서브클래싱** 하여
- 결과 반영 로직(``_apply_result``), 순서 계산(``_build_ordered_universe``),
@@ -36,17 +41,21 @@ from __future__ import annotations
import json
import random
import re
import threading
import time
from typing import Any, Dict, List, Optional, Set
from .condition_manager import ConditionSearchManager
from ..utils.env import get_env_float, get_env_from_db, get_env_int
from ..utils.env import get_env_bool, get_env_float, get_env_from_db, get_env_int
from ..utils.logger import get_logger
from ..ws.kis_ws import _get_kiwoom_token_cached
logger = get_logger("kis_trader.kwcond")
# 키움 CNSRREQ rc=900003 — WS Bye 후에도 계정에 조건검색 등록이 남는 경우
_CNSR_ALREADY_REGISTERED_RC = "900003"
def _normalize_code(raw) -> str:
"""키움 종목코드 정규화: 'A005930' → '005930' (KIS 6자리 코드계와 정합)."""
@@ -59,6 +68,22 @@ def _normalize_code(raw) -> str:
return c
def _extract_cnsr_seq(data: Dict) -> str:
"""CNSRREQ/CNSRCLR 응답에서 seq 추출.
실패 응답은 top-level ``seq`` 가 비고 ``return_msg`` 에만
``...(seq=1)`` 형태로 오는 경우가 있음 (문서·실제 필드 불일치).
"""
seq = str(data.get("seq") or "").strip()
if seq:
return seq
msg = str(data.get("return_msg") or "")
m = re.search(r"seq\s*=\s*(\d+)", msg, flags=re.IGNORECASE)
if m:
return m.group(1).strip()
return ""
class KiwoomConditionSearchManager(ConditionSearchManager):
"""키움 웹소켓 실시간 조건검색 매니저.
@@ -133,6 +158,10 @@ class KiwoomConditionSearchManager(ConditionSearchManager):
self._cnsrreq_pending: Set[str] = set()
self._cnsrreq_confirmed: Set[str] = set()
self._cnsrreq_retry_timer: Optional[threading.Timer] = None
# 900003(이미 등록) 에 대한 CNSRCLR→재요청은 seq당 사이클 1회만 (무한 재시도 금지)
self._cnsr_dup_cleared: Set[str] = set()
# CNSRCLR→REQ 파이프라인에서 send 대상 WS 캐시
self._cnsr_send_ws: Any = None
# ------------------------------------------------------------------
# Public API (오버라이드) — 부모 start() 는 REST 폴링이므로 사용 안 함
@@ -258,11 +287,20 @@ class KiwoomConditionSearchManager(ConditionSearchManager):
self._shared_ws.send_json({"trnm": "CNSRLST"})
def _on_shared_login(self, ws) -> None:
"""시세 WS 재접속마다 조건식 목록 재조회 → CNSRREQ 재등록."""
"""시세 WS 재접속마다 조건식 목록 재조회 → (CLR 후) CNSRREQ 재등록.
키움은 새벽 등 WS ``Bye`` 후에도 계정 단위로 조건검색 등록이 남을 수 있음.
재LOGIN 시에는 기존 pending 재시도 타이머를 끊고, CNSRLST→CLR→REQ 로 맞춘다.
"""
if not self._running:
return
self._cancel_cnsrreq_retry()
self._ready.clear()
self._start_ok = False
with self._kw_lock:
self._cnsr_dup_cleared.clear()
self._cnsrreq_pending.clear()
self._cnsrreq_confirmed.clear()
try:
ws.send(json.dumps({"trnm": "CNSRLST"}))
logger.debug("키움 조건검색 [공유WS] LOGIN → CNSRLST")
@@ -276,7 +314,12 @@ class KiwoomConditionSearchManager(ConditionSearchManager):
elif trnm == "CNSRREQ":
self._handle_cnsrreq(msg)
elif trnm == "CNSRCLR":
logger.debug("키움 조건검색 CNSRCLR 응답: rc=%s", msg.get("return_code"))
logger.debug(
"키움 조건검색 CNSRCLR 응답: seq=%s rc=%s msg=%s",
_extract_cnsr_seq(msg),
msg.get("return_code"),
msg.get("return_msg"),
)
def _on_shared_real(self, ws, msg: dict) -> None:
"""조건검색 편입/이탈 REAL — 843 필드 있는 항목만 처리."""
@@ -369,7 +412,12 @@ class KiwoomConditionSearchManager(ConditionSearchManager):
return
if trnm == "CNSRCLR":
logger.debug("키움 조건검색 CNSRCLR 응답: rc=%s", data.get("return_code"))
logger.debug(
"키움 조건검색 CNSRCLR 응답: seq=%s rc=%s msg=%s",
_extract_cnsr_seq(data),
data.get("return_code"),
data.get("return_msg"),
)
return
# ------------------------------------------------------------------
@@ -387,7 +435,7 @@ class KiwoomConditionSearchManager(ConditionSearchManager):
return "", ""
def _handle_condition_list(self, ws, rows: List) -> None:
"""CNSRLST 응답으로 name→seq 해결 후, 전략별 seq 확정 + CNSRREQ 발송."""
"""CNSRLST 응답으로 name→seq 해결 후, 전략별 seq 확정 + CLR→CNSRREQ."""
name_to_seq: Dict[str, str] = {}
for it in rows:
seq, name = self._seq_name(it)
@@ -429,9 +477,55 @@ class KiwoomConditionSearchManager(ConditionSearchManager):
self._sid_by_seq = sid_by_seq
self._active_seqs = list(sid_by_seq.keys())
# 실시간(search_type=1) 등록 — seq 별 순차 발송 (레이트리밋·응답 누락 방지)
# 실시간(search_type=1) 등록 — 유령등록 방지로 CLR 후 순차 REQ
self._send_cnsrreq_all(ws)
def _cancel_cnsrreq_retry(self) -> None:
if self._cnsrreq_retry_timer:
try:
self._cnsrreq_retry_timer.cancel()
except Exception:
pass
self._cnsrreq_retry_timer = None
def _send_one_cnsrclr(self, ws, seq: str) -> bool:
"""단일 seq CNSRCLR(실시간 해제) 발송 — Bye 후 유령 등록 제거용."""
payload = {"trnm": "CNSRCLR", "seq": seq}
try:
if self._shared_mode and self._shared_ws:
return bool(self._shared_ws.send_json(payload))
target = ws if ws is not None else self._ws
if target is None:
return False
target.send(json.dumps(payload))
return True
except Exception as e:
logger.warning("키움 CNSRCLR 발송 실패 (seq=%s): %s", seq, e)
return False
def _clear_active_seqs(self, ws, seqs: Optional[List[str]] = None) -> None:
"""active(또는 주어진) seq 전부 CNSRCLR — 간격은 env (하드코딩 금지)."""
if seqs is None:
with self._kw_lock:
seqs = list(self._active_seqs)
else:
seqs = list(seqs)
try:
seqs.sort(key=lambda x: int(x) if str(x).isdigit() else 0)
except Exception:
pass
if not seqs:
return
gap_lo = float(get_env_float("KIWOOM_CNSRCLR_GAP_MIN_SEC", 0.3))
gap_hi = float(get_env_float("KIWOOM_CNSRCLR_GAP_MAX_SEC", 0.6))
if gap_hi < gap_lo:
gap_lo, gap_hi = gap_hi, gap_lo
logger.info("🧹 키움 조건검색 CNSRCLR 선행 (seq=%s)", seqs)
for i, seq in enumerate(seqs):
if i > 0:
time.sleep(random.uniform(gap_lo, gap_hi))
self._send_one_cnsrclr(ws, seq)
def _send_one_cnsrreq(self, ws, seq: str) -> bool:
"""단일 seq CNSRREQ 발송."""
payload = {
@@ -450,14 +544,24 @@ class KiwoomConditionSearchManager(ConditionSearchManager):
return False
def _send_cnsrreq_all(self, ws) -> None:
"""active seq 목록을 간격 두고 순차 CNSRREQ — 미응답 seq 는 타이머로 재발송."""
"""active seq: (옵션) CNSRCLR → settle → 순차 CNSRREQ. 미응답만 유한 재시도."""
self._cancel_cnsrreq_retry()
self._cnsr_send_ws = ws
with self._kw_lock:
seqs = list(self._active_seqs)
self._cnsr_dup_cleared.clear()
try:
seqs.sort(key=lambda x: int(x))
except Exception:
pass
# Bye 후 유령 등록(900003) 방지: 재등록 전에 항상 클리어 (기본 ON)
if get_env_bool("KIWOOM_CNSRCLR_BEFORE_REQ", True):
self._clear_active_seqs(ws, seqs)
settle = float(get_env_float("KIWOOM_CNSRCLR_SETTLE_SEC", 1.0))
if settle > 0:
time.sleep(settle)
gap_lo = float(get_env_float("KIWOOM_CNSRREQ_GAP_MIN_SEC", 0.8))
gap_hi = float(get_env_float("KIWOOM_CNSRREQ_GAP_MAX_SEC", 1.5))
if gap_hi < gap_lo:
@@ -480,7 +584,7 @@ class KiwoomConditionSearchManager(ConditionSearchManager):
self._schedule_cnsrreq_retry(ws, attempt=1)
def _schedule_cnsrreq_retry(self, ws, *, attempt: int) -> None:
"""CNSRREQ 응답이 안 온 seq 만 간격 두고 재발송."""
"""CNSRREQ 응답이 안 온 seq 만 간격 두고 재발송 (상한 초과 시 중단 · 무한 금지)."""
max_retries = get_env_int("KIWOOM_CNSRREQ_MAX_RETRIES", 3)
retry_delay = float(get_env_float("KIWOOM_CNSRREQ_RETRY_SEC", 5.0))
gap_lo = float(get_env_float("KIWOOM_CNSRREQ_GAP_MIN_SEC", 0.8))
@@ -488,12 +592,7 @@ class KiwoomConditionSearchManager(ConditionSearchManager):
if gap_hi < gap_lo:
gap_lo, gap_hi = gap_hi, gap_lo
if self._cnsrreq_retry_timer:
try:
self._cnsrreq_retry_timer.cancel()
except Exception:
pass
self._cnsrreq_retry_timer = None
self._cancel_cnsrreq_retry()
def _retry() -> None:
if not self._running:
@@ -507,9 +606,13 @@ class KiwoomConditionSearchManager(ConditionSearchManager):
return
if attempt > max_retries:
logger.warning(
"⚠️ 키움 CNSRREQ 미응답 seq=%s — 최대 재시도 초과",
"⚠️ 키움 CNSRREQ 미응답 seq=%s — 최대 재시도 초과 (추가 재시도 없음)",
missing,
)
# pending 비워 이후 타이머/재발송이 남지 않게 함
with self._kw_lock:
for s in missing:
self._cnsrreq_pending.discard(s)
return
logger.warning(
"⚠️ 키움 CNSRREQ 미응답 seq=%s → %.0fs 후 재발송 (%d/%d)",
@@ -525,21 +628,63 @@ class KiwoomConditionSearchManager(ConditionSearchManager):
self._cnsrreq_retry_timer.daemon = True
self._cnsrreq_retry_timer.start()
def _handle_cnsr_already_registered(self, seq: str, data: Dict) -> None:
"""rc=900003: seq당 1회만 CNSRCLR→CNSRREQ. 두 번째면 포기(무한 금지)."""
with self._kw_lock:
already = seq in self._cnsr_dup_cleared
if not already:
self._cnsr_dup_cleared.add(seq)
if already:
logger.warning(
"⚠️ 키움 CNSRREQ 900003 재발 (seq=%s) — CLR 후에도 충돌, 추가 재시도 없음 msg=%s",
seq, data.get("return_msg"),
)
with self._kw_lock:
self._cnsrreq_pending.discard(seq)
return
ws = self._cnsr_send_ws
if ws is None and self._shared_mode:
ws = self._shared_ws
if ws is None:
ws = self._ws
logger.warning(
"⚠️ 키움 CNSRREQ 이미등록(900003) seq=%s → CNSRCLR 1회 후 재요청",
seq,
)
self._send_one_cnsrclr(ws, seq)
settle = float(get_env_float("KIWOOM_CNSRCLR_SETTLE_SEC", 1.0))
if settle > 0:
time.sleep(settle)
self._send_one_cnsrreq(ws, seq)
def _handle_cnsrreq(self, data: Dict) -> None:
"""CNSRREQ 초기 응답: 현재 매칭 종목 리스트로 seq universe 초기화."""
rc = str(data.get("return_code"))
seq = str(data.get("seq") or "").strip()
if rc not in ("0", "0.0"):
logger.warning(
"키움 CNSRREQ 실패 (seq=%s) rc=%s msg=%s",
seq, rc, data.get("return_msg"),
)
return
seq = _extract_cnsr_seq(data)
# seq 가 응답에 없을 수 있음 → active_seqs 가 1개면 그걸로 간주
if not seq:
with self._kw_lock:
if len(self._active_seqs) == 1:
seq = self._active_seqs[0]
if rc not in ("0", "0.0"):
msg = str(data.get("return_msg") or "")
is_dup = (
rc == _CNSR_ALREADY_REGISTERED_RC
or "이미 등록" in msg
)
logger.warning(
"키움 CNSRREQ 실패 (seq=%s) rc=%s msg=%s",
seq, rc, msg,
)
if is_dup and seq:
self._handle_cnsr_already_registered(seq, data)
return
# 그 외 에러: pending 에서 제거 → '미응답'으로 오인·무한 재발송 방지
if seq:
with self._kw_lock:
self._cnsrreq_pending.discard(seq)
return
if not seq:
return
@@ -569,13 +714,20 @@ class KiwoomConditionSearchManager(ConditionSearchManager):
if not code:
continue
ins_del = str(vals.get("843") or "").strip().upper()
# seq 는 item 또는 상위에서 옴. item 에 없으면 유일 seq 로 폴백.
seq = str(it.get("item") or it.get("seq") or "").strip() if isinstance(it, dict) else ""
# 조건식 번호는 values.841. item 은 종목코드(9001 과 동일)인 경우가 많아 seq 로 쓰면 안 됨.
seq = str(vals.get("841") or "").strip()
if not seq and isinstance(it, dict):
raw_item = str(it.get("seq") or "").strip()
if raw_item and raw_item in self._sid_by_seq:
seq = raw_item
if not seq:
if len(self._active_seqs) == 1:
seq = self._active_seqs[0]
else:
# seq 불명 + 다중 조건 → 어느 유니버스인지 특정 불가, skip
logger.debug(
"키움 조건 REAL skip — seq 미식별 code=%s 843=%s active=%s",
code, ins_del, self._active_seqs,
)
continue
bucket = self._seq_codes.setdefault(seq, {})
if ins_del == "D":
@@ -609,7 +761,16 @@ class KiwoomConditionSearchManager(ConditionSearchManager):
bucket = dict(self._seq_codes.get(seq, {}))
sids = list(self._sid_by_seq.get(seq, []))
# rows: 삽입순(HTS 응답 순서) 유지 → _build_ordered_universe 가 그대로 사용
rows = [{"code": c, "name": n} for c, n in bucket.items()]
rows = []
for c, n in bucket.items():
disp = str(n or c).strip() or c
if disp == c and self.db is not None:
try:
from ..utils.stock_name import resolve_stock_display_name
disp = resolve_stock_display_name(self.db, c, fallback=c)
except Exception:
disp = c
rows.append({"code": c, "name": disp})
from ..utils.universe_source import universe_source_active
for sid in sids:
if not universe_source_active(sid, "kiwoom_condition"):