125 lines
4.2 KiB
Python
125 lines
4.2 KiB
Python
"""
|
|
kis_trader 전략 ID 단일 정의 — active_trades / trade_history / 웹 실거래·보유탭 공통.
|
|
|
|
- ``main.py`` 전략 클래스의 ``strategy_id`` (SCALP, SHORT, MOMENTUM, UPDOW, BREAKOUT, HOLDING)
|
|
- ``trade_history`` 신규 기록은 ``canonical_strategy_id()`` 로 위 ID 만 저장 → 실거래 분석 탭과 일치
|
|
- 구식명(SCALP_RSI_REVERSAL, SHORT_ANT_SHAKING)은 조회 시 접두어(LIKE)로 묶고, 신규 저장은 canonical
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
from typing import List, Optional
|
|
|
|
# kis_trader/main.py 에 등록되는 전략 ID (실거래 분석·보유탭 공통)
|
|
# US_MOMENTUM 은 MOMENTUM 접두 접힘 방지용 — canonical 에서 먼저 매칭
|
|
KIS_TRADER_STRATEGY_IDS: List[str] = [
|
|
"SCALP",
|
|
"SHORT",
|
|
"MOMENTUM",
|
|
"US_MOMENTUM",
|
|
"UPDOW",
|
|
"BREAKOUT",
|
|
"RANGE_BREAK",
|
|
"DBBAND",
|
|
"HOLDING",
|
|
]
|
|
|
|
# 사용자 지정 — 웹·계산에서 완전히 숨길 전략 (비활성·중복·저성능).
|
|
# DB 에 기존 거래기록이 남아 있어도 실거래 분석·대시보드·보유탭·운영설정에서 표시·집계하지 않는다.
|
|
# RANGE_BREAK : BREAKOUT 과 돌파 컨셉 중복 + 미검증 (파라서치 거래 0건)
|
|
# DBBAND : 표본 부족·미성숙 (백테 2~4건)
|
|
# ※ SCALP 는 실거래 분석·대시보드에 노출 (스캘핑 봇 운영 중)
|
|
HIDDEN_STRATEGY_IDS: List[str] = [
|
|
"RANGE_BREAK",
|
|
"DBBAND",
|
|
]
|
|
|
|
# 웹·실거래 집계에서 제외 (DB에 남아 있어도 표시·집계 안 함)
|
|
EXCLUDED_STRATEGY_IDS: List[str] = [
|
|
"MANUAL",
|
|
"SCALP_RSI_REVERSAL",
|
|
"SHORT_ANT_SHAKING",
|
|
"SCALP_TEST",
|
|
] + HIDDEN_STRATEGY_IDS
|
|
|
|
# 보유·매도 탭 전용 — 홀딩봇(HOLDING)은 조회만, 여기서 매도 대상 아님
|
|
PORTFOLIO_EXCLUDED_STRATEGY_IDS: List[str] = EXCLUDED_STRATEGY_IDS + [
|
|
"HOLDING",
|
|
]
|
|
|
|
|
|
def canonical_strategy_id(strategy: Optional[str]) -> str:
|
|
"""
|
|
DB 저장·trade_history 기록용 canonical ID.
|
|
MANUAL 은 그대로, 구식명은 SCALP/SHORT 등으로 접힘.
|
|
"""
|
|
s = (strategy or "").strip().upper()
|
|
if not s:
|
|
return "MANUAL"
|
|
if s == "MANUAL":
|
|
return "MANUAL"
|
|
# 해외 모멘텀 — startswith("MOMENTUM") 에 잡히지 않도록 먼저 처리
|
|
if s == "US_MOMENTUM" or s.startswith("US_MOMENTUM"):
|
|
return "US_MOMENTUM"
|
|
if s.startswith("SCALP"):
|
|
return "SCALP"
|
|
if s.startswith("SHORT") or s.startswith("TAIL"):
|
|
return "SHORT"
|
|
if s.startswith("MOMENTUM"):
|
|
return "MOMENTUM"
|
|
if s.startswith("UPDOW"):
|
|
return "UPDOW"
|
|
if s.startswith("BREAKOUT"):
|
|
return "BREAKOUT"
|
|
if s.startswith("RANGE_BREAK"):
|
|
return "RANGE_BREAK"
|
|
if s.startswith("DBBAND") or s.startswith("BBBAND"):
|
|
return "DBBAND"
|
|
if s.startswith("HOLDING"):
|
|
return "HOLDING"
|
|
return s
|
|
|
|
|
|
def strategy_prefix_for_filter(strategy: Optional[str]) -> str:
|
|
"""
|
|
trade_history / active_trades LIKE 필터용 접두어.
|
|
실거래 분석 탭·보유탭이 동일 규칙 사용.
|
|
"""
|
|
s_upper = (strategy or "").upper().strip()
|
|
if not s_upper or s_upper == "ALL":
|
|
return ""
|
|
if s_upper == "MANUAL":
|
|
return "MANUAL"
|
|
if s_upper == "US_MOMENTUM" or s_upper.startswith("US_MOMENTUM"):
|
|
return "US_MOMENTUM"
|
|
if s_upper.startswith("SCALP"):
|
|
return "SCALP"
|
|
if s_upper.startswith("SHORT") or s_upper.startswith("TAIL"):
|
|
return "SHORT"
|
|
if s_upper.startswith("MOMENTUM"):
|
|
return "MOMENTUM"
|
|
if s_upper.startswith("UPDOW"):
|
|
return "UPDOW"
|
|
if s_upper.startswith("BREAKOUT"):
|
|
return "BREAKOUT"
|
|
if s_upper.startswith("RANGE_BREAK"):
|
|
return "RANGE_BREAK"
|
|
if s_upper.startswith("DBBAND") or s_upper.startswith("BBBAND"):
|
|
return "DBBAND"
|
|
if s_upper.startswith("HOLDING"):
|
|
return "HOLDING"
|
|
return s_upper
|
|
|
|
|
|
def strategy_like_pattern(strategy: Optional[str]) -> Optional[str]:
|
|
"""None 이면 전체, 아니면 ``PREFIX%``."""
|
|
prefix = strategy_prefix_for_filter(strategy)
|
|
if not prefix:
|
|
return None
|
|
return prefix + "%"
|
|
|
|
|
|
def is_bot_strategy(strategy: Optional[str]) -> bool:
|
|
"""kis_trader 봇 전략 row 인지 (MANUAL 제외)."""
|
|
c = canonical_strategy_id(strategy)
|
|
return c in KIS_TRADER_STRATEGY_IDS
|