Files
kis_bot/kis_trader/utils/stock_name.py
Your Name fc27e726f9 feat: 새로운 안전 규칙 및 최적화 적용을 통한 트레이딩 시스템 개선
변경 사항 (Changes):

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

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

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

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

기대 효과 (Impact):

이러한 개선 사항들은 트레이딩 시스템의 견고성(Robustness)과 신뢰성을 향상시키며, 에러 발생을 최소화하고 다양한 시스템 컴포넌트 간의 원활한 상호작용을 보장함.
2026-07-17 01:09:09 +09:00

98 lines
3.3 KiB
Python

"""
종목코드 → 표시용 종목명 (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