import sys
sys.path.append('../../../')

from bfxapi import Client, PUB_WS_HOST, PUB_REST_HOST

# Retrieving orderbook requires public hosts
bfx = Client(
  manageOrderBooks=True,
  ws_host=PUB_WS_HOST,
  rest_host=PUB_REST_HOST
)

@bfx.ws.on('error')
def log_error(err):
  print ("Error: {}".format(err))

@bfx.ws.on('unsubscribed')
async def on_unsubscribe(subscription):
  print ("Unsubscribed from {}".format(subscription.symbol))
  # await subscription.subscribe()

@bfx.ws.on('subscribed')
async def on_subscribe(subscription):
  print ("Subscribed to {}".format(subscription.symbol))
  # await subscription.unsubscribe()
  # or
  # await bfx.ws.unsubscribe(subscription.chanId)

@bfx.ws.once('subscribed')
async def on_once_subscribe(subscription):
  print ("Performig resubscribe")
  await bfx.ws.resubscribe(subscription.chan_id)


async def start():
  await bfx.ws.subscribe('book', 'tBTCUSD')

bfx.ws.on('connected', start)
bfx.ws.run()