from database import TradeDB

db = TradeDB()
try:
    snap = db.get_merged_env_snapshot()
    updated = False
    
    # 전략별 후보 상한(CAND_LIMIT)을 0(무제한)으로 설정
    for prefix in ["SCALP", "SHORT", "MOMENTUM", "BREAKOUT", "RANGE_BREAK", "UPDOW", "DART", "US_MOMENTUM"]:
        key = f"{prefix}_CAND_LIMIT"
        if snap.get(key) != "0":
            snap[key] = "0"
            updated = True
            print(f"Set {key} to 0 (Unlimited)")
            
    if updated:
        db.insert_env_snapshot(snap)
        print("Inserted new env snapshot with CAND_LIMIT removed.")
    else:
        print("All CAND_LIMITs are already 0.")
finally:
    db.close()