Files
kis_bot/scratch/db_check4.py

39 lines
1.3 KiB
Python

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()