feat(tests): 신규 키움 웹소켓 조건검색 및 실시간 조건검색 테스트 추가
변경 사항 ---- - _test_kiwoom_condition_list.py: 키움 웹소켓 조건검색 '목록조회' 기능을 단독으로 테스트하는 스크립트 추가 - _test_kiwoom_condition_realtime.py: 'momentum' 조건식을 실시간으로 등록하고 초기 매칭 종목 리스트 및 실시간 편입/이탈을 수신하는 테스트 스크립트 추가 - _verify_columnar_bitid.py, _verify_shared_e2e_breakout.py, _verify_shared_e2e.py: 공유 메모리 및 dict 간의 데이터 일관성을 검증하는 테스트 추가 영향 ---- - 신규 테스트 스크립트 추가로 키움 웹소켓 API의 기능 검증 및 안정성을 높임 - 기존 기능에 대한 영향 없음 Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
420
kis_trader/engine/daily_profit_halt.py
Normal file
420
kis_trader/engine/daily_profit_halt.py
Normal file
@@ -0,0 +1,420 @@
|
||||
"""
|
||||
kis_trader/engine/daily_profit_halt.py — 일일 익절 목표 달성 시 신규 매수 중단 (2단)
|
||||
==============================================================================
|
||||
1단(마스터): ``DAILY_PROFIT_*`` — 봇 당일 실현손익 총합 → 전 전략 매수 OFF
|
||||
2단(서브) : ``{STRATEGY}_DAILY_PROFIT_*`` — 해당 전략만 매수 OFF
|
||||
|
||||
목표 판정: 원(KRW) · 운용한도 대비 %(PCT) **둘 중 하나라도** 달성 시 트리거.
|
||||
분모(%) : ``DAILY_PROFIT_TARGET_BUDGET_KRW`` 또는 ON 전략 ``*_TOTAL_BUDGET_KRW`` 합.
|
||||
|
||||
매도(손절·익절)는 계속 — **신규 매수만** 차단.
|
||||
모든 임계값 env/DB — 하드코딩 금지.
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
import threading
|
||||
import time
|
||||
from datetime import datetime as dt
|
||||
from typing import Any, Callable, Dict, List, Optional, Tuple
|
||||
|
||||
from kis_trader.utils.env import get_env_bool, get_env_float, get_env_from_db, get_env_int
|
||||
from kis_trader.utils.logger import get_logger
|
||||
|
||||
logger = get_logger("kis_trader.daily_profit_halt")
|
||||
|
||||
# 전략 ID → 운용한도 env 키 (bot_pct 분모와 동일)
|
||||
_STRATEGY_BUDGET_ENV: Dict[str, str] = {
|
||||
"SCALP": "SCALP_TOTAL_BUDGET_KRW",
|
||||
"SHORT": "TAIL_TOTAL_BUDGET_KRW",
|
||||
"MOMENTUM": "MOMENTUM_TOTAL_BUDGET_KRW",
|
||||
"BREAKOUT": "BREAKOUT_TOTAL_BUDGET_KRW",
|
||||
"RANGE_BREAK": "RANGE_BREAK_TOTAL_BUDGET_KRW",
|
||||
"UPDOW": "UPDOW_TOTAL_BUDGET_KRW",
|
||||
"DBBAND": "DBBAND_TOTAL_BUDGET_KRW",
|
||||
}
|
||||
|
||||
|
||||
def _strategy_prefix(strategy_id: str) -> str:
|
||||
return (strategy_id or "").strip().upper() or "STRATEGY"
|
||||
|
||||
|
||||
def _enabled_flag(common_key: str, strategy_key: str, default: bool = False) -> bool:
|
||||
raw = str(get_env_from_db(strategy_key, "")).strip()
|
||||
if raw != "":
|
||||
return get_env_bool(strategy_key, default)
|
||||
return get_env_bool(common_key, default)
|
||||
|
||||
|
||||
def _target_krw(common_key: str, strategy_key: str, default: float = 0.0) -> float:
|
||||
sk = strategy_key
|
||||
if str(get_env_from_db(sk, "")).strip():
|
||||
return max(0.0, float(get_env_float(sk, default)))
|
||||
return max(0.0, float(get_env_float(common_key, default)))
|
||||
|
||||
|
||||
def _target_pct(common_key: str, strategy_key: str, default: float = 0.0) -> float:
|
||||
sk = strategy_key
|
||||
if str(get_env_from_db(sk, "")).strip():
|
||||
return max(0.0, float(get_env_float(sk, default)))
|
||||
return max(0.0, float(get_env_float(common_key, default)))
|
||||
|
||||
|
||||
def _mode(common_key: str, strategy_key: str, default: str = "fixed") -> str:
|
||||
"""익절 판정 방식 — 전략 서브값 우선 → 마스터 → 기본(fixed).
|
||||
|
||||
fixed : 기존 동작(고정 목표 KRW/PCT 도달 시 중단)
|
||||
trailing: 당일 손익 고점 대비 되돌림(trail) 발동
|
||||
both : 고정(하드캡) OR 트레일 — 먼저 닿는 쪽
|
||||
"""
|
||||
raw = str(get_env_from_db(strategy_key, "")).strip()
|
||||
if raw != "":
|
||||
return raw.lower()
|
||||
g = str(get_env_from_db(common_key, "")).strip()
|
||||
return (g or default).lower()
|
||||
|
||||
|
||||
def load_global_profit_target() -> Dict[str, Any]:
|
||||
return {
|
||||
"enabled": get_env_bool("DAILY_PROFIT_TARGET_ENABLED", False),
|
||||
"krw": max(0.0, float(get_env_float("DAILY_PROFIT_TARGET_KRW", 0.0))),
|
||||
"pct": max(0.0, float(get_env_float("DAILY_PROFIT_TARGET_PCT", 0.0))),
|
||||
"budget_krw": max(0.0, float(get_env_float("DAILY_PROFIT_TARGET_BUDGET_KRW", 0.0))),
|
||||
"halt_new_buys": get_env_bool("DAILY_PROFIT_HALT_NEW_BUYS", True),
|
||||
"notify_mm": get_env_bool("DAILY_PROFIT_NOTIFY_MM", True),
|
||||
# 트레일링 익절 (당일 손익 고점 추적) — fixed 기본이라 미설정 시 동작 불변
|
||||
"mode": _mode("DAILY_PROFIT_MODE", "DAILY_PROFIT_MODE", "fixed"),
|
||||
# 다단계 트레일 tier(우선) — '수익원:컷%,…'. 비면 단일 drop_pct 폴백.
|
||||
"trail_tiers": str(get_env_from_db("DAILY_PROFIT_TRAIL_TIERS", "")).strip(),
|
||||
"trail_drop_pct": max(0.0, float(get_env_float("DAILY_PROFIT_TRAIL_DROP_PCT", 0.0))),
|
||||
"trail_arm_krw": max(0.0, float(get_env_float("DAILY_PROFIT_TRAIL_ARM_KRW", 0.0))),
|
||||
"trail_arm_pct": max(0.0, float(get_env_float("DAILY_PROFIT_TRAIL_ARM_PCT", 0.0))),
|
||||
}
|
||||
|
||||
|
||||
def load_strategy_profit_target(strategy_id: str) -> Dict[str, Any]:
|
||||
pfx = _strategy_prefix(strategy_id)
|
||||
return {
|
||||
"enabled": _enabled_flag(
|
||||
"DAILY_PROFIT_TARGET_ENABLED",
|
||||
f"{pfx}_DAILY_PROFIT_TARGET_ENABLED",
|
||||
False,
|
||||
),
|
||||
"krw": _target_krw(
|
||||
"DAILY_PROFIT_TARGET_KRW",
|
||||
f"{pfx}_DAILY_PROFIT_TARGET_KRW",
|
||||
0.0,
|
||||
),
|
||||
"pct": _target_pct(
|
||||
"DAILY_PROFIT_TARGET_PCT",
|
||||
f"{pfx}_DAILY_PROFIT_TARGET_PCT",
|
||||
0.0,
|
||||
),
|
||||
"budget_env": _STRATEGY_BUDGET_ENV.get(pfx, f"{pfx}_TOTAL_BUDGET_KRW"),
|
||||
"halt_new_buys": get_env_bool(f"{pfx}_DAILY_PROFIT_HALT_NEW_BUYS", True),
|
||||
# 트레일링 익절 — 전략 서브값 우선 → 마스터 폴백 (fixed 기본)
|
||||
"mode": _mode("DAILY_PROFIT_MODE", f"{pfx}_DAILY_PROFIT_MODE", "fixed"),
|
||||
# 다단계 트레일 tier — 전략 서브값 우선 → 마스터 폴백 (비면 단일 drop_pct)
|
||||
"trail_tiers": _mode("DAILY_PROFIT_TRAIL_TIERS", f"{pfx}_DAILY_PROFIT_TRAIL_TIERS", ""),
|
||||
"trail_drop_pct": _target_krw(
|
||||
"DAILY_PROFIT_TRAIL_DROP_PCT", f"{pfx}_DAILY_PROFIT_TRAIL_DROP_PCT", 0.0,
|
||||
),
|
||||
"trail_arm_krw": _target_krw(
|
||||
"DAILY_PROFIT_TRAIL_ARM_KRW", f"{pfx}_DAILY_PROFIT_TRAIL_ARM_KRW", 0.0,
|
||||
),
|
||||
"trail_arm_pct": _target_pct(
|
||||
"DAILY_PROFIT_TRAIL_ARM_PCT", f"{pfx}_DAILY_PROFIT_TRAIL_ARM_PCT", 0.0,
|
||||
),
|
||||
}
|
||||
|
||||
|
||||
def resolve_strategy_budget_krw(strategy_id: str) -> float:
|
||||
pfx = _strategy_prefix(strategy_id)
|
||||
env_key = _STRATEGY_BUDGET_ENV.get(pfx, f"{pfx}_TOTAL_BUDGET_KRW")
|
||||
v = get_env_int(env_key, 0)
|
||||
if v > 0:
|
||||
return float(v)
|
||||
# SLOT × MAX_STOCKS 폴백
|
||||
slot_key = f"{pfx}_SLOT_MONEY"
|
||||
max_key = f"{pfx}_MAX_STOCKS"
|
||||
slot = get_env_int(slot_key, 0)
|
||||
mx = get_env_int(max_key, 0)
|
||||
if slot > 0 and mx > 0:
|
||||
return float(slot * mx)
|
||||
return 0.0
|
||||
|
||||
|
||||
def resolve_global_operating_budget_krw(active_strategy_ids: List[str]) -> float:
|
||||
explicit = max(0.0, float(get_env_float("DAILY_PROFIT_TARGET_BUDGET_KRW", 0.0)))
|
||||
if explicit > 0:
|
||||
return explicit
|
||||
total = 0.0
|
||||
for sid in active_strategy_ids:
|
||||
b = resolve_strategy_budget_krw(sid)
|
||||
if b > 0:
|
||||
total += b
|
||||
return total
|
||||
|
||||
|
||||
def _target_configured(cfg: Dict[str, Any]) -> bool:
|
||||
if not cfg.get("enabled"):
|
||||
return False
|
||||
return float(cfg.get("krw") or 0) > 0 or float(cfg.get("pct") or 0) > 0
|
||||
|
||||
|
||||
def _target_reached(pnl_krw: float, cfg: Dict[str, Any], budget_krw: float) -> bool:
|
||||
if not _target_configured(cfg):
|
||||
return False
|
||||
krw_tgt = float(cfg.get("krw") or 0)
|
||||
pct_tgt = float(cfg.get("pct") or 0)
|
||||
if krw_tgt > 0 and pnl_krw >= krw_tgt:
|
||||
return True
|
||||
if pct_tgt > 0 and budget_krw > 0:
|
||||
need = budget_krw * pct_tgt / 100.0
|
||||
if pnl_krw >= need:
|
||||
return True
|
||||
return False
|
||||
|
||||
|
||||
def _guard_active(cfg: Dict[str, Any]) -> bool:
|
||||
"""모드별 가드 작동 여부 — fixed:목표값 / trailing:trail_drop / both:둘 중 하나.
|
||||
|
||||
(기존 _target_configured 는 fixed 전용으로 유지하고, buy_allowed 는 이 함수를 쓴다.)
|
||||
"""
|
||||
if not cfg.get("enabled"):
|
||||
return False
|
||||
mode = str(cfg.get("mode") or "fixed").lower()
|
||||
if mode in ("fixed", "both"):
|
||||
if float(cfg.get("krw") or 0) > 0 or float(cfg.get("pct") or 0) > 0:
|
||||
return True
|
||||
if mode in ("trailing", "both"):
|
||||
if parse_trail_tiers(cfg.get("trail_tiers")):
|
||||
return True
|
||||
if float(cfg.get("trail_drop_pct") or 0) > 0:
|
||||
return True
|
||||
return False
|
||||
|
||||
|
||||
def _trail_arm_krw(cfg: Dict[str, Any], budget_krw: float) -> float:
|
||||
"""트레일 발동 최소수익(원) — KRW·PCT 중 큰 값. 0 이면 트레일 비활성(과민발동 방지)."""
|
||||
arm = float(cfg.get("trail_arm_krw") or 0)
|
||||
arm_pct = float(cfg.get("trail_arm_pct") or 0)
|
||||
if arm_pct > 0 and budget_krw > 0:
|
||||
arm = max(arm, budget_krw * arm_pct / 100.0)
|
||||
return arm
|
||||
|
||||
|
||||
def parse_trail_tiers(raw: Any) -> List[Tuple[float, float]]:
|
||||
"""다단계 트레일 tier 파싱: '수익원:컷%,…' → [(arm_krw, drop_pct), …] (arm 오름차순).
|
||||
|
||||
개별포지션 ratchet_tiers 와 동일 철학을 당일 누적손익에 적용.
|
||||
예) '30000:50,100000:35,300000:25'
|
||||
= 누적 3만↑ 고점대비 -50% / 10만↑ -35% / 30만↑ -25% (벌수록 타이트)
|
||||
잘못된 토큰은 건너뜀. 빈 문자열 → [].
|
||||
"""
|
||||
if raw is None:
|
||||
return []
|
||||
s = str(raw).strip()
|
||||
if s == "" or s.lower() == "off":
|
||||
return []
|
||||
tiers: List[Tuple[float, float]] = []
|
||||
for tok in s.split(","):
|
||||
tok = tok.strip()
|
||||
if not tok or ":" not in tok:
|
||||
continue
|
||||
a, _, d = tok.partition(":")
|
||||
try:
|
||||
arm = float(a.strip())
|
||||
drop = float(d.strip())
|
||||
except (TypeError, ValueError):
|
||||
continue
|
||||
if arm < 0 or drop <= 0:
|
||||
continue
|
||||
tiers.append((arm, drop))
|
||||
tiers.sort(key=lambda x: x[0])
|
||||
return tiers
|
||||
|
||||
|
||||
def _active_tier_drop(peak_krw: float, tiers: List[Tuple[float, float]]) -> float:
|
||||
"""peak 가 도달한 가장 높은 tier 의 컷% 반환. 최저 tier 미만이면 0(비발동)."""
|
||||
drop = 0.0
|
||||
for arm, d in tiers: # arm 오름차순
|
||||
if peak_krw >= arm:
|
||||
drop = d
|
||||
else:
|
||||
break
|
||||
return drop
|
||||
|
||||
|
||||
def _trail_reached(
|
||||
pnl_krw: float, peak_krw: float, cfg: Dict[str, Any], budget_krw: float,
|
||||
) -> bool:
|
||||
"""당일 손익 고점(peak) 대비 되돌림 시 트리거.
|
||||
|
||||
우선순위: 다단계 tier(trail_tiers) → 단일 trail_drop_pct(하위호환).
|
||||
tier: peak 가 속한 구간의 컷% 적용 → 벌수록 타이트(이익 보존 강화).
|
||||
안전장치: tier·drop 없거나 ARM 미달이면 비활성(설정 없이 과민발동 금지).
|
||||
peak 가 ARM 이상으로 올라간 뒤부터만 감시 → 작은 수익에 조기 종료 방지.
|
||||
"""
|
||||
# 1) 다단계 tier 우선
|
||||
tiers = parse_trail_tiers(cfg.get("trail_tiers"))
|
||||
if tiers:
|
||||
drop = _active_tier_drop(peak_krw, tiers)
|
||||
if drop <= 0:
|
||||
return False # 최저 tier(=arm) 미달
|
||||
cut = peak_krw * (1.0 - drop / 100.0)
|
||||
return pnl_krw <= cut
|
||||
|
||||
# 2) 단일 drop_pct 폴백 (기존 동작)
|
||||
drop = float(cfg.get("trail_drop_pct") or 0)
|
||||
if drop <= 0:
|
||||
return False
|
||||
arm = _trail_arm_krw(cfg, budget_krw)
|
||||
if arm <= 0:
|
||||
return False
|
||||
if peak_krw < arm:
|
||||
return False
|
||||
cut = peak_krw * (1.0 - drop / 100.0)
|
||||
return pnl_krw <= cut
|
||||
|
||||
|
||||
def _format_hit_detail(
|
||||
pnl_krw: float, cfg: Dict[str, Any], budget_krw: float,
|
||||
) -> str:
|
||||
parts = [f"실현 {pnl_krw:+,.0f}원"]
|
||||
krw_tgt = float(cfg.get("krw") or 0)
|
||||
pct_tgt = float(cfg.get("pct") or 0)
|
||||
if krw_tgt > 0:
|
||||
parts.append(f"목표 {krw_tgt:,.0f}원")
|
||||
if pct_tgt > 0 and budget_krw > 0:
|
||||
parts.append(f"목표 {pct_tgt:.2f}% (한도 {budget_krw:,.0f}원)")
|
||||
return " · ".join(parts)
|
||||
|
||||
|
||||
class DailyProfitHaltGuard:
|
||||
"""
|
||||
Orchestrator 가 주입 — ``buy_allowed(strategy_id)`` 로 신규 매수 차단 여부 판단.
|
||||
"""
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
*,
|
||||
global_pnl_fn: Callable[[str], Tuple[float, int]],
|
||||
strategy_pnl_fn: Callable[[str, str], Tuple[float, int]],
|
||||
active_strategies_fn: Callable[[], List[str]],
|
||||
notify_fn: Optional[Callable[[str, Optional[str]], None]] = None,
|
||||
):
|
||||
self._global_pnl_fn = global_pnl_fn
|
||||
self._strategy_pnl_fn = strategy_pnl_fn
|
||||
self._active_strategies_fn = active_strategies_fn
|
||||
self._notify_fn = notify_fn
|
||||
self._lock = threading.Lock()
|
||||
self._notified_keys: set = set()
|
||||
self._last_log_ts: Dict[str, float] = {}
|
||||
# 트레일링용 당일 손익 고점 추적 {scope_key: (today, peak_krw)}
|
||||
self._peaks: Dict[str, Tuple[str, float]] = {}
|
||||
|
||||
def _update_peak(self, scope_key: str, today: str, pnl_krw: float) -> float:
|
||||
"""당일 손익 고점 갱신·반환. 날짜가 바뀌면 리셋."""
|
||||
with self._lock:
|
||||
day, peak = self._peaks.get(scope_key, (today, pnl_krw))
|
||||
if day != today:
|
||||
peak = pnl_krw
|
||||
peak = max(peak, pnl_krw)
|
||||
self._peaks[scope_key] = (today, peak)
|
||||
return peak
|
||||
|
||||
def buy_allowed(self, strategy_id: str) -> Tuple[bool, str]:
|
||||
"""신규 매수 허용 여부. (False, 사유) 이면 매수 스킵."""
|
||||
today = dt.now().strftime("%Y-%m-%d")
|
||||
sid = _strategy_prefix(strategy_id)
|
||||
|
||||
gcfg = load_global_profit_target()
|
||||
if gcfg.get("halt_new_buys", True) and _guard_active(gcfg):
|
||||
gpnl, gcnt = self._global_pnl_fn(today)
|
||||
gbudget = resolve_global_operating_budget_krw(self._active_strategies_fn())
|
||||
gpeak = self._update_peak(f"global:{today}", today, gpnl)
|
||||
hit, extra = self._check_hit(gpnl, gpeak, gcfg, gbudget)
|
||||
if hit:
|
||||
detail = _format_hit_detail(gpnl, gcfg, gbudget) + extra
|
||||
self._notify_once(f"global:{today}", "GLOBAL", detail, gcnt)
|
||||
self._throttled_log(sid, f"⛔ [일일익절·총합] {detail} → 신규매수 중단")
|
||||
return False, "탈락-일일익절(총합)"
|
||||
|
||||
scfg = load_strategy_profit_target(sid)
|
||||
if scfg.get("halt_new_buys", True) and _guard_active(scfg):
|
||||
spnl, scnt = self._strategy_pnl_fn(today, sid)
|
||||
sbudget = resolve_strategy_budget_krw(sid)
|
||||
speak = self._update_peak(f"{sid}:{today}", today, spnl)
|
||||
hit, extra = self._check_hit(spnl, speak, scfg, sbudget)
|
||||
if hit:
|
||||
detail = _format_hit_detail(spnl, scfg, sbudget) + extra
|
||||
self._notify_once(f"{sid}:{today}", sid, detail, scnt)
|
||||
self._throttled_log(
|
||||
sid,
|
||||
f"⛔ [일일익절·{sid}] {detail} → 해당전략 신규매수 중단",
|
||||
)
|
||||
return False, f"탈락-일일익절({sid})"
|
||||
|
||||
return True, ""
|
||||
|
||||
def _check_hit(
|
||||
self, pnl_krw: float, peak_krw: float, cfg: Dict[str, Any], budget_krw: float,
|
||||
) -> Tuple[bool, str]:
|
||||
"""모드별 트리거 판정. (hit, 로그 부가설명) 반환.
|
||||
|
||||
both: 고정(하드캡) 먼저 검사 → 트레일 — 먼저 닿는 쪽 발동.
|
||||
"""
|
||||
mode = str(cfg.get("mode") or "fixed").lower()
|
||||
if mode in ("fixed", "both") and _target_reached(pnl_krw, cfg, budget_krw):
|
||||
return True, ""
|
||||
if mode in ("trailing", "both") and _trail_reached(pnl_krw, peak_krw, cfg, budget_krw):
|
||||
tiers = parse_trail_tiers(cfg.get("trail_tiers"))
|
||||
drop = (
|
||||
_active_tier_drop(peak_krw, tiers) if tiers
|
||||
else float(cfg.get("trail_drop_pct") or 0)
|
||||
)
|
||||
tag = "트레일·다단" if tiers else "트레일"
|
||||
return True, (
|
||||
f" · {tag}(고점 {peak_krw:+,.0f}원 대비 -{drop:.0f}% 되돌림)"
|
||||
)
|
||||
return False, ""
|
||||
|
||||
def _throttled_log(self, strategy_id: str, msg: str) -> None:
|
||||
now = time.time()
|
||||
key = strategy_id
|
||||
if now - self._last_log_ts.get(key, 0.0) < 60.0:
|
||||
return
|
||||
self._last_log_ts[key] = now
|
||||
logger.info(msg)
|
||||
|
||||
def _notify_once(
|
||||
self, notify_key: str, scope: str, detail: str, trade_cnt: int,
|
||||
) -> None:
|
||||
gcfg = load_global_profit_target()
|
||||
if not gcfg.get("notify_mm", True) or not self._notify_fn:
|
||||
return
|
||||
with self._lock:
|
||||
if notify_key in self._notified_keys:
|
||||
return
|
||||
self._notified_keys.add(notify_key)
|
||||
title = "🎯 **일일 익절 목표 달성**"
|
||||
if scope == "GLOBAL":
|
||||
body = (
|
||||
f"{title}\n"
|
||||
f"- 범위: **전체 봇 (총합)**\n"
|
||||
f"- {detail}\n"
|
||||
f"- 청산 {trade_cnt}건\n"
|
||||
f"- 조치: **금일 신규 매수 중단** (보유 종목 매도·손절은 유지)"
|
||||
)
|
||||
else:
|
||||
body = (
|
||||
f"{title}\n"
|
||||
f"- 범위: **{scope}**\n"
|
||||
f"- {detail}\n"
|
||||
f"- 청산 {trade_cnt}건\n"
|
||||
f"- 조치: **{scope} 신규 매수만 중단** (다른 전략·보유 매도는 유지)"
|
||||
)
|
||||
try:
|
||||
self._notify_fn(body, scope if scope != "GLOBAL" else None)
|
||||
except Exception as ex:
|
||||
logger.debug("일일익절 MM 알림 실패: %s", ex)
|
||||
Reference in New Issue
Block a user