shuim
408bcce62e
감사에서 드러난 신뢰성·조용한실패 버그 수정. 전략값(§5)·시그니처(§4) 불변. ① KIS API 신뢰성(korea_investment_api, config): 스로틀에 _throttle_lock 추가로 thundering herd(8코루틴 동시발사) 제거, API_TIMEOUT 10→20초, circuit_breaker 임계 15→7, 재시도 백오프 지터. → 시간당 121건 실패의 근원 완화. ② portfolio summary 보존(portfolio): 잔고 API 실패 시 summary가 영영 None으로 남아 현금 0→매수 스킵되던 것을, 최초 실패 시 holdings 기반 부분 summary로 채움. ③ output2 빈리스트 가드: KIS가 output2=[] 반환 시 IndexError로 MDD 업데이트가 조용히 스킵(하방보호 비활성)되던 것 방어. swing/scheduler·adaptive_strategy (실버그) + daytrading 2곳(정규화). trade_executor:555는 load-bearing 가드라 의도적 미변경. ④ MIN_BUY_AMOUNT config화+소액완화(trade_executor, config): 하드코딩 50만을 MIN_BUY_AMOUNT_KRW로. 자산 833만↓ 계좌가 전종목 차단되던 것을 자산 대비로 완화(큰 계좌는 기존 50만 하한 불변). ⑤ adaptive 서브버그(adaptive_strategy): fallback 시 _last_update_ts 갱신(refresh 남발 방지), force_mode_override 전체필드(crisis의 max_daily_trades=0 실제 적용), get_event_loop→get_running_loop. 검증: code-verifier 통과(CONFIRMED 버그/회귀 없음, 4개 위험지점 안전), pytest 442 통과. (day_trading.py·entry_service.py는 소량 인접 선재편집 포함 — 검증 완료.) Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01HBmzf1HDnvFHLraJ8eaaKJ
266 lines
12 KiB
Python
266 lines
12 KiB
Python
"""스윙 트레이딩 독립 스케줄러
|
|
|
|
main.py의 lifespan에서 asyncio.create_task()로 시작되는
|
|
백그라운드 태스크. 기존 auto_trade_scheduler()를 대체.
|
|
|
|
60초 주기 실행 흐름:
|
|
1. 장 오픈 체크 (장외 시 30분 대기)
|
|
2. 스윙 포지션 손절/익절 체크 (check_and_execute_exit_v2)
|
|
3. MDD 체크 + 강제 축소 실행 (10% → 30% 축소, 20% → 50% 축소)
|
|
4. AUTO_TRADE_ENABLED 체크
|
|
5. Redis 캐시 또는 DB에서 추천 조회
|
|
6. MCP RISK_ASSESSMENT → 승인 시 스윙 매수 실행
|
|
|
|
MDD 축소 (CLAUDE.md 5.3):
|
|
- reduction_applied == "10pct": 30% 축소 (REDUCE_AND_PAUSE)
|
|
- reduction_applied == "20pct": 50% 축소 (FORCE_REDUCE, 레버리지 ETF 우선)
|
|
|
|
추천 데이터 소스 (우선순위):
|
|
1. Redis: "recommendations:latest" 키 (배치 분석 결과 캐시)
|
|
2. DB: get_rule_based_recommendations_from_db() (Redis 미스 시 폴백)
|
|
"""
|
|
import asyncio
|
|
import logging
|
|
|
|
import pytz
|
|
from datetime import timedelta as _td
|
|
|
|
from app.core.config import settings
|
|
from app.core.database import AsyncSessionLocal
|
|
from app.core.market_schedule import is_market_open
|
|
from app.services.mdd_manager import mdd_manager
|
|
from app.services.swing.exit_service import check_and_execute_exit_v2
|
|
from app.services.swing.entry_service import execute_swing_entries, scan_dual_strategy_signals
|
|
|
|
logger = logging.getLogger(__name__)
|
|
_KST = pytz.timezone("Asia/Seoul")
|
|
|
|
|
|
def _normalize_recommendations(rec: dict) -> dict:
|
|
"""recommendations_by_rule 키 정규화.
|
|
|
|
Redis 캐시에 "recommendations" 키로 저장된 경우
|
|
"recommendations_by_rule"로 변환하여 execute_swing_entries()와 호환.
|
|
|
|
Args:
|
|
rec: 추천 딕셔너리
|
|
|
|
Returns:
|
|
dict: 정규화된 추천 딕셔너리
|
|
"""
|
|
if not isinstance(rec, dict):
|
|
return rec
|
|
if "recommendations_by_rule" in rec:
|
|
return rec
|
|
if "recommendations" in rec:
|
|
rec["recommendations_by_rule"] = rec["recommendations"]
|
|
return rec
|
|
|
|
|
|
def _filter_unsuitable_stocks(rec: dict, unsuitable_codes: set) -> None:
|
|
"""장 전 AI 분석에서 부적합 판정된 종목을 추천 목록에서 제거."""
|
|
groups = rec.get("recommendations_by_rule", [])
|
|
if isinstance(groups, list):
|
|
for group in groups:
|
|
if isinstance(group, dict) and "stocks" in group:
|
|
group["stocks"] = [
|
|
s for s in group["stocks"]
|
|
if isinstance(s, dict) and s.get("stock_code") not in unsuitable_codes
|
|
]
|
|
elif isinstance(groups, dict):
|
|
for key, stocks in groups.items():
|
|
if isinstance(stocks, list):
|
|
groups[key] = [
|
|
s for s in stocks
|
|
if isinstance(s, dict) and s.get("stock_code") not in unsuitable_codes
|
|
]
|
|
|
|
|
|
async def swing_scheduler() -> None:
|
|
"""60초 주기 스윙 트레이딩 스케줄러.
|
|
|
|
무한 루프로 실행되며 asyncio.CancelledError로만 종료.
|
|
각 사이클에서 예외 발생 시 로그 기록 후 다음 사이클 계속.
|
|
|
|
DB 세션은 사이클마다 AsyncSessionLocal()로 새로 생성하여
|
|
장시간 세션 유지로 인한 커넥션 풀 고갈 방지.
|
|
"""
|
|
await asyncio.sleep(2)
|
|
from app.api.endpoints.recommendations import (
|
|
get_rule_based_recommendations_from_db,
|
|
)
|
|
from app.api.endpoints.mcp import request_mcp_risk_assessment
|
|
from app.core.redis import get_json
|
|
|
|
logger.info("📈 [Swing] Scheduler started (every 60s)")
|
|
try:
|
|
while True:
|
|
try:
|
|
logger.info("[Swing] ── 사이클 시작 ──")
|
|
|
|
if not is_market_open():
|
|
logger.info("[Swing] 장 미오픈 — 30분 대기")
|
|
await asyncio.sleep(1800)
|
|
continue
|
|
|
|
# ── Phase 1: 청산 체크 (별도 세션) ──
|
|
async with AsyncSessionLocal() as exit_session:
|
|
try:
|
|
exits = await check_and_execute_exit_v2(exit_session)
|
|
if exits:
|
|
logger.info(f"[Swing] 청산 실행: {len(exits)}건")
|
|
except Exception as e:
|
|
logger.error(f"[Swing] 청산 체크 오류: {e}", exc_info=True)
|
|
|
|
# ── Phase 2: MDD 업데이트 + 체크 + 강제 축소 ──
|
|
# [FIX] mdd_manager.update()를 매 사이클 호출하여 MDD 감지 활성화
|
|
try:
|
|
from app.services.korea_investment_api import korea_investment_api
|
|
balance_resp = await korea_investment_api.get_balance()
|
|
if balance_resp:
|
|
_total = float((balance_resp.get("output2") or [{}])[0].get("tot_evlu_amt", 0))
|
|
if _total > 0:
|
|
mdd_result = await mdd_manager.update(_total)
|
|
if mdd_result.get("action") != "NONE":
|
|
logger.warning(f"[Swing] MDD 상태 변경: {mdd_result}")
|
|
except Exception as e:
|
|
logger.debug(f"[Swing] MDD update 조회 실패 (무시): {e}")
|
|
|
|
can_open_new = await mdd_manager.can_open_new_position()
|
|
logger.info(f"[Swing] MDD 체크: can_open={can_open_new}")
|
|
if not can_open_new:
|
|
logger.warning("[Swing] MDD 초과 — 신규 진입 차단")
|
|
if mdd_manager.reduction_applied:
|
|
try:
|
|
reduce_pct = 0.50 if mdd_manager.reduction_applied == "20pct" else 0.30
|
|
mdd_action = {
|
|
"action": "FORCE_REDUCE" if mdd_manager.reduction_applied == "20pct" else "REDUCE_AND_PAUSE",
|
|
"reduce_pct": reduce_pct,
|
|
"prioritize_leverage": mdd_manager.reduction_applied == "20pct",
|
|
}
|
|
results = await mdd_manager.execute_reduction(mdd_action)
|
|
if results:
|
|
logger.warning(f"[Swing] MDD 강제 축소 완료: {len(results)}종목")
|
|
except Exception as e:
|
|
logger.error(f"[Swing] MDD 강제 축소 실패: {e}", exc_info=True)
|
|
await asyncio.sleep(60)
|
|
continue
|
|
|
|
# 3. 자동매매 활성화 여부
|
|
if not settings.AUTO_TRADE_ENABLED:
|
|
logger.info("[Swing] AUTO_TRADE_ENABLED=False — 스킵")
|
|
await asyncio.sleep(60)
|
|
continue
|
|
|
|
# ── Phase 3: 추천 조회 + 진입 실행 (별도 세션) ──
|
|
# 4. Redis 캐시 확인 → cache miss 시 DB 조회
|
|
rec = await get_json("recommendations:latest")
|
|
# [FIX] Redis 캐시가 존재해도 recommendations_by_rule이 비어있으면(배치가 빈
|
|
# 추천을 캐싱한 경우) 기존엔 '히트'로 처리해 DB 폴백을 건너뛰고 스킵했다.
|
|
# 빈 캐시가 30분 TTL 동안 DB의 유효 후보를 가려 매매를 막으므로, 비어있으면
|
|
# 캐시 미스와 동일하게 DB 조회로 폴백한다.
|
|
if rec and rec.get("recommendations_by_rule"):
|
|
rec = _normalize_recommendations(rec)
|
|
logger.info(f"[Swing] Redis 캐시 히트 (추천 {len(rec.get('recommendations_by_rule', []))}그룹)")
|
|
else:
|
|
logger.info("[Swing] Redis 캐시 미스/빈값 → DB 조회")
|
|
async with AsyncSessionLocal() as rec_session:
|
|
rec = await get_rule_based_recommendations_from_db(rec_session)
|
|
|
|
if not rec or not rec.get("recommendations_by_rule"):
|
|
logger.info("[Swing] 추천 데이터 없음 — 스킵")
|
|
await asyncio.sleep(60)
|
|
continue
|
|
|
|
# 5. 장 전 AI 분석 결과 활용 + MCP 리스크 평가 (최적화)
|
|
# 장 전 Sonnet 배치 분석 결과가 Redis에 있으면 부적합 종목을 사전 필터링
|
|
# MCP RISK_ASSESSMENT는 실제 매수 직전에만 호출 (비용 절감)
|
|
_ai_filtered_codes = set()
|
|
try:
|
|
_ai_pre = await get_json("ai:pre_market:latest")
|
|
if _ai_pre and isinstance(_ai_pre, dict):
|
|
for s in _ai_pre.get("suitable_stocks", []):
|
|
if isinstance(s, dict) and not s.get("suitable", True):
|
|
_ai_filtered_codes.add(s.get("stock_code", ""))
|
|
if _ai_filtered_codes:
|
|
logger.info(f"[Swing] AI 장 전 분석 필터: {len(_ai_filtered_codes)}종목 제외")
|
|
except Exception:
|
|
pass
|
|
|
|
# 추천 목록에서 AI 부적합 종목 제거
|
|
if _ai_filtered_codes:
|
|
_filter_unsuitable_stocks(rec, _ai_filtered_codes)
|
|
|
|
recommendations = rec.get("recommendations_by_rule", [])
|
|
if isinstance(recommendations, dict):
|
|
rec_count = sum(len(v) for v in recommendations.values())
|
|
top_stocks = [
|
|
item.get("stock_code", "")
|
|
for items in recommendations.values()
|
|
for item in (items[:3] if isinstance(items, list) else [])
|
|
][:10]
|
|
else:
|
|
rec_count = sum(
|
|
len(g.get("stocks", [])) for g in recommendations
|
|
if isinstance(g, dict)
|
|
)
|
|
top_stocks = [
|
|
s.get("stock_code", "")
|
|
for g in recommendations if isinstance(g, dict)
|
|
for s in g.get("stocks", [])[:3]
|
|
][:10]
|
|
|
|
if not top_stocks:
|
|
logger.info("[Swing] 필터링 후 추천 종목 없음 — 스킵")
|
|
await asyncio.sleep(60)
|
|
continue
|
|
|
|
# MCP 리스크 평가 (Haiku — 실제 진입 직전 1회)
|
|
trade_plan = {
|
|
"action": "BUY",
|
|
"strategy": "swing",
|
|
"total_recommendations": rec_count,
|
|
"top_stocks": top_stocks,
|
|
}
|
|
|
|
logger.info(f"[Swing] MCP 리스크 평가 요청: {len(top_stocks)}종목")
|
|
mcp_approved, mcp_reason = await request_mcp_risk_assessment(
|
|
trade_plan, {"strategy": "swing"}, {"top_stocks": top_stocks}
|
|
)
|
|
logger.info(f"[Swing] MCP 결과: approved={mcp_approved}, reason={mcp_reason}")
|
|
if not mcp_approved:
|
|
logger.warning(f"[Swing] MCP 거부: {mcp_reason}")
|
|
from app.services.db_worker import enqueue_db_write
|
|
await enqueue_db_write({
|
|
"type": "log_mcp_deny",
|
|
"reason": mcp_reason,
|
|
"trade_plan": trade_plan,
|
|
})
|
|
await asyncio.sleep(60)
|
|
continue
|
|
|
|
# 6. 스윙 진입 실행 (별도 세션)
|
|
async with AsyncSessionLocal() as entry_session:
|
|
logger.info(f"[Swing] 진입 실행 시작: {top_stocks[:5]}")
|
|
await execute_swing_entries(rec, entry_session)
|
|
|
|
# 7. 듀얼 전략 시그널 스캔 (모멘텀 돌파 + 눌림목 매수, CLAUDE.md 8.4)
|
|
try:
|
|
async with AsyncSessionLocal() as dual_session:
|
|
dual_results = await scan_dual_strategy_signals(dual_session)
|
|
if dual_results:
|
|
logger.info(f"[Swing] 듀얼 전략 진입: {len(dual_results)}건")
|
|
except Exception as e:
|
|
logger.debug(f"[Swing] 듀얼 전략 스캔 오류: {e}")
|
|
|
|
logger.info("[Swing] ── 사이클 완료 ──")
|
|
|
|
except Exception as e:
|
|
logger.error(f"[Swing] 스케줄러 사이클 오류: {e}", exc_info=True)
|
|
|
|
await asyncio.sleep(60)
|
|
|
|
except asyncio.CancelledError:
|
|
logger.info("[Swing] Scheduler stopped")
|
|
raise
|