"""
LS 데이터 누적 중단 및 KIS 검증 모드 설정 스크립트
"""
import sys
sys.path.insert(0, "/home/hoon/kis_bot")
from database import TradeDB
from kis_trader.utils.env import invalidate_merged_env_cache

db = TradeDB()

try:
    # 현재 스냅샷 가져오기
    snap = dict(db.get_merged_env_snapshot() or {})
    
    changed = False
    
    # 1. LS 데이터 누적 중단 (사용자 요청: 조건식 연결 안 하면 안 쌓이게)
    # LS 관련 환경변수를 False로 변경
    ls_keys = [
        "LS_WS_VALIDATION_ENABLED",
        "LS_CONDITION_HISTORY_ENABLED",
        "LS_WS_FOLLOW_CONDITION_UNIVERSE",
        "LS_WS_ALSO_HOGA"
    ]
    
    print("=== 기존 LS 설정 ===")
    for k in ls_keys:
        print(f"  {k}: {snap.get(k)}")
        if str(snap.get(k)).lower() == 'true' or snap.get(k) is True:
            snap[k] = False
            changed = True
            
    # 2. KIS 시세 구독 비교를 위해 WS_PROVIDER 변경
    # (이미 kis_with_validation일 수 있지만 확실히 함)
    current_provider = snap.get("WS_PROVIDER")
    print(f"\n=== 기존 WS_PROVIDER ===")
    print(f"  WS_PROVIDER: {current_provider}")
    
    if current_provider != "kis_with_validation":
        snap["WS_PROVIDER"] = "kis_with_validation"
        changed = True
        
    if changed:
        print("\n=== 설정 변경 적용 중... ===")
        db.insert_env_snapshot(snap)
        invalidate_merged_env_cache()
        print("✅ DB 환경변수 업데이트 및 캐시 무효화 완료!")
    else:
        print("\n✅ 이미 설정이 올바르게 반영되어 있어 변경할 내용이 없습니다.")

finally:
    db.close()