43 lines
1.0 KiB
Python
43 lines
1.0 KiB
Python
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()
|