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

def check_db():
    db = TradeDB()
    try:
        th_cols = [dict(r)["Field"] for r in db.conn.execute("SHOW COLUMNS FROM trade_history").fetchall()]
        print("trade_history cols:", th_cols)
        
        at_cols = [dict(r)["Field"] for r in db.conn.execute("SHOW COLUMNS FROM active_trades").fetchall()]
        print("active_trades cols:", at_cols)
        
        env_cols = [dict(r)["Field"] for r in db.conn.execute("SHOW COLUMNS FROM env_config").fetchall()]
        print("env_config cols:", env_cols)
        
        query = "SELECT * FROM env_config"
        envs = db.conn.execute(query).fetchall()
        for e in envs:
            print(dict(e))

        today = datetime.now().strftime("%Y-%m-%d")
        print(f"--- Trades for {today} ---")
        trades = db.conn.execute("SELECT * FROM trade_history WHERE DATE(buy_time) = %s", (today,)).fetchall()
        for t in trades:
            print(dict(t))
            
        print(f"--- Active Trades for {today} ---")
        active_trades = db.conn.execute("SELECT * FROM active_trades WHERE DATE(buy_time) = %s", (today,)).fetchall()
        for t in active_trades:
            print(dict(t))

    finally:
        db.close()

if __name__ == "__main__":
    check_db()