import sys
import json
sys.path.append('/home/hoon/kis_bot')
from database import TradeDB

def check_db():
    db = TradeDB()
    try:
        # Check last 5 momentum trades
        trades = db.conn.execute(
            "SELECT code, name, env_snapshot FROM trade_history WHERE strategy='MOMENTUM' ORDER BY id DESC LIMIT 5"
        ).fetchall()
        for t in trades:
            print(f"Code: {t['code']}, Name: {t['name']}")
            if t['env_snapshot']:
                try:
                    snap = json.loads(t['env_snapshot'])
                    ob = snap.get("orderbook", {})
                    print("  Orderbook snapshot:", ob)
                    print("  Entry features:", snap.get("entry_features", {}))
                except Exception as e:
                    print("  Error parsing JSON:", e)
            else:
                print("  env_snapshot is NULL")
    finally:
        db.close()

if __name__ == "__main__":
    check_db()