import time
from okx_trader import OKXTrader
import json


def main():
    trader = OKXTrader()
    instrument = "BTC-USDT"

    trader.subscribe_candlesticks(instrument, timeframe="5m")
    # trader.subscribe_trades(instrument)
    trader.subscribe_ticker(instrument)
    # trader.subscribe_book(instrument, depth=5)
    # trader.subscribe_user_order()
    # trader.subscribe_user_trade()
    # trader.subscribe_user_balance()
 
    # print(trader.get_balance())

    try:
        while True:
            try:
                data = trader.ws.recv()
            except Exception as e:
                print(f"[WARN] WebSocket disconnected or error: {e}.")
            
                trader = OKXTrader()
                trader.subscribe_ticker(instrument)
            
                continue

            if data == '': continue

            msg = json.loads(data)
            print(json.dumps(msg, indent=4))
            
            time.sleep(1)
    except KeyboardInterrupt:
        print('Exiting...')

if __name__ == '__main__':
    main()