#!/usr/bin/env python3
"""
DB에서 계좌번호 설정 확인 스크립트
- 현재 최신 env_config에서 계좌번호 관련 필드 조회
- 모의/실전 구분 및 실제 읽히는 값 확인
"""
import sys
from pathlib import Path

SCRIPT_DIR = Path(__file__).resolve().parent
sys.path.insert(0, str(SCRIPT_DIR))

from database import TradeDB

def main():
    db = TradeDB(db_path=str(SCRIPT_DIR / "quant_bot.db"))
    env_data = db.get_latest_env()
    db.close()

    if not env_data or not env_data.get("snapshot"):
        print("❌ env_config에 데이터가 없습니다.")
        return

    snapshot = env_data["snapshot"]
    print(f"📊 최신 env_config (id={env_data['id']}, created_at={env_data['created_at']})\n")

    # 모의 여부
    kis_mock = snapshot.get("KIS_MOCK", "").lower()
    is_mock = kis_mock in ("true", "1", "yes")
    print(f"🔹 KIS_MOCK = '{kis_mock}' → 모의투자: {is_mock}\n")

    # 실전 계좌
    print("📌 실전 계좌 (KIS_MOCK=false 시 사용):")
    acc_no_real = snapshot.get("KIS_ACCOUNT_NO", "").strip()
    acc_code_real = snapshot.get("KIS_ACCOUNT_CODE", "").strip() or "01"
    print(f"  KIS_ACCOUNT_NO = '{acc_no_real}' ({len(acc_no_real)}자리)")
    print(f"  KIS_ACCOUNT_CODE = '{acc_code_real}' ({len(acc_code_real)}자리)")
    if acc_no_real:
        digits_no = "".join(c for c in acc_no_real if c.isdigit())
        print(f"  → 숫자만 추출: '{digits_no}' ({len(digits_no)}자리)")
        if len(digits_no) >= 10:
            print(f"  → CANO: '{digits_no[:8]}', ACNT_PRDT_CD: '{digits_no[8:10]}' ✅")
        elif len(digits_no) == 8:
            print(f"  → CANO: '{digits_no}', ACNT_PRDT_CD: '{acc_code_real.zfill(2)[:2]}' ✅")
        else:
            print(f"  → ⚠️ 8자리 미만: CANO='{digits_no.zfill(8)[:8]}', ACNT_PRDT_CD='{acc_code_real.zfill(2)[:2]}' (부족한 자리 0으로 채움)")
    else:
        print("  → ❌ 값이 비어있음!")

    print()

    # 모의 계좌
    print("📌 모의 계좌 (KIS_MOCK=true 시 사용):")
    acc_no_mock = snapshot.get("KIS_ACCOUNT_NO_MOCK", "").strip()
    acc_code_mock = snapshot.get("KIS_ACCOUNT_CODE_MOCK", "").strip() or "01"
    print(f"  KIS_ACCOUNT_NO_MOCK = '{acc_no_mock}' ({len(acc_no_mock)}자리)")
    print(f"  KIS_ACCOUNT_CODE_MOCK = '{acc_code_mock}' ({len(acc_code_mock)}자리)")
    if acc_no_mock:
        digits_no = "".join(c for c in acc_no_mock if c.isdigit())
        print(f"  → 숫자만 추출: '{digits_no}' ({len(digits_no)}자리)")
        if len(digits_no) >= 10:
            print(f"  → CANO: '{digits_no[:8]}', ACNT_PRDT_CD: '{digits_no[8:10]}' ✅")
        elif len(digits_no) == 8:
            print(f"  → CANO: '{digits_no}', ACNT_PRDT_CD: '{acc_code_mock.zfill(2)[:2]}' ✅")
        else:
            print(f"  → ⚠️ 8자리 미만: CANO='{digits_no.zfill(8)[:8]}', ACNT_PRDT_CD='{acc_code_mock.zfill(2)[:2]}' (부족한 자리 0으로 채움)")
    else:
        print("  → ❌ 값이 비어있음! (fallback: 실전 계좌 사용)")

    print()

    # 실제 사용될 계좌
    print("🎯 실제 사용될 계좌:")
    if is_mock:
        final_no = acc_no_mock or acc_no_real
        final_code = acc_code_mock if acc_no_mock else acc_code_real
        print(f"  모의투자 모드 → KIS_ACCOUNT_NO_MOCK='{acc_no_mock}' 또는 KIS_ACCOUNT_NO='{acc_no_real}'")
    else:
        final_no = acc_no_real
        final_code = acc_code_real
        print(f"  실전투자 모드 → KIS_ACCOUNT_NO='{acc_no_real}'")
    
    if final_no:
        digits_no = "".join(c for c in final_no if c.isdigit())
        if len(digits_no) >= 10:
            cano = digits_no[:8]
            acnt = digits_no[8:10]
        elif len(digits_no) == 8:
            cano = digits_no
            acnt = final_code.zfill(2)[:2]
        else:
            cano = digits_no.zfill(8)[:8]
            acnt = final_code.zfill(2)[:2]
        
        if len(cano) == 8 and len(acnt) == 2:
            print(f"  ✅ 최종: CANO={cano}, ACNT_PRDT_CD={acnt}")
        else:
            print(f"  ❌ 최종: CANO={cano}({len(cano)}자리), ACNT_PRDT_CD={acnt}({len(acnt)}자리) → OPSQ2000 발생 가능!")
    else:
        print("  ❌ 계좌번호가 비어있음 → OPSQ2000 발생!")

if __name__ == "__main__":
    main()