feat: 새로운 안전 규칙 및 최적화 적용을 통한 트레이딩 시스템 개선
변경 사항 (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:
97
kis_trader/utils/stock_name.py
Normal file
97
kis_trader/utils/stock_name.py
Normal file
@@ -0,0 +1,97 @@
|
||||
"""
|
||||
종목코드 → 표시용 종목명 (Mattermost·로그·DB 공통).
|
||||
|
||||
키움 조건검색 실시간 응답에는 종목명이 없어 code=이름 임시값이 들어오는 경우가 많다.
|
||||
우선순위: 유효 fallback → 메모리 캐시 → stock_meta → target_candidates_history → 잔고 prdt_name
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
import logging
|
||||
from typing import Dict, Optional
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
# 프로세스 내 재조회 방지 (매수체크 루프·체결 알림 공용)
|
||||
_NAME_CACHE: Dict[str, str] = {}
|
||||
|
||||
|
||||
def resolve_stock_display_name(
|
||||
db,
|
||||
code: str,
|
||||
fallback: str = "",
|
||||
*,
|
||||
holdings_map: Optional[Dict[str, Dict]] = None,
|
||||
cache_to_meta: bool = True,
|
||||
) -> str:
|
||||
"""
|
||||
종목코드에 대응하는 한글 종목명을 반환한다.
|
||||
|
||||
Args:
|
||||
db: TradeDB / TradeDBExt 인스턴스
|
||||
code: 6자리 종목코드
|
||||
fallback: 전략·후보에서 넘어온 이름 (코드와 같으면 무시)
|
||||
holdings_map: {code: {name, qty, ...}} — OrderManager 잔고 캐시 등
|
||||
cache_to_meta: history/잔고에서 찾은 이름을 stock_meta 에 저장
|
||||
"""
|
||||
code = str(code or "").strip()
|
||||
if not code:
|
||||
return str(fallback or "").strip()
|
||||
|
||||
fb = str(fallback or "").strip()
|
||||
if fb and fb != code:
|
||||
_NAME_CACHE[code] = fb
|
||||
return fb
|
||||
|
||||
cached = _NAME_CACHE.get(code)
|
||||
if cached and cached != code:
|
||||
return cached
|
||||
|
||||
# 1) stock_meta
|
||||
try:
|
||||
if hasattr(db, "get_stock_meta"):
|
||||
meta = db.get_stock_meta(code)
|
||||
if meta:
|
||||
n = str(meta.get("name") or "").strip()
|
||||
if n and n != code:
|
||||
_NAME_CACHE[code] = n
|
||||
return n
|
||||
except Exception as exc:
|
||||
logger.debug("get_stock_meta 실패(%s): %s", code, exc)
|
||||
|
||||
# 2) target_candidates_history (조건검색 스냅샷 — 실매 후보 적재)
|
||||
try:
|
||||
conn = getattr(db, "conn", None)
|
||||
if conn is not None:
|
||||
row = conn.execute(
|
||||
"SELECT name FROM target_candidates_history "
|
||||
"WHERE code = %s AND name IS NOT NULL AND name != '' AND name != code "
|
||||
"ORDER BY COALESCE(event_time, scan_time) DESC, id DESC LIMIT 1",
|
||||
(code,),
|
||||
).fetchone()
|
||||
if row:
|
||||
n = str(row.get("name") or "").strip()
|
||||
if n and n != code:
|
||||
_NAME_CACHE[code] = n
|
||||
if cache_to_meta and hasattr(db, "upsert_stock_meta"):
|
||||
try:
|
||||
db.upsert_stock_meta(code, name=n)
|
||||
except Exception:
|
||||
pass
|
||||
return n
|
||||
except Exception as exc:
|
||||
logger.debug("target_candidates_history 이름 조회 실패(%s): %s", code, exc)
|
||||
|
||||
# 3) 실계좌 잔고 (보유 종목 prdt_name)
|
||||
if holdings_map:
|
||||
h = holdings_map.get(code) or {}
|
||||
n = str(h.get("name") or "").strip()
|
||||
if n and n != code:
|
||||
_NAME_CACHE[code] = n
|
||||
if cache_to_meta and hasattr(db, "upsert_stock_meta"):
|
||||
try:
|
||||
db.upsert_stock_meta(code, name=n)
|
||||
except Exception:
|
||||
pass
|
||||
return n
|
||||
|
||||
return fb if fb else code
|
||||
Reference in New Issue
Block a user