Rewrite, edit and organize examples/websocket demos.

examples/websocket/public/derivatives_status.py

@@ -0,0 +1,23 @@
+# python -c "import examples.websocket.public.derivatives_status"
+
+from bfxapi import Client, PUB_WSS_HOST
+from bfxapi.websocket.enums import Error, Channel
+from bfxapi.websocket.types import DerivativesStatus
+
+from bfxapi.websocket import subscriptions
+
+bfx = Client(WSS_HOST=PUB_WSS_HOST)
+
+@bfx.wss.on("derivatives_status_update")
+def on_derivatives_status_update(subscription: subscriptions.Status, data: DerivativesStatus):
+  print(f"{subscription}:", data)
+
+@bfx.wss.on("wss-error")
+def on_wss_error(code: Error, msg: str):
+    print(code, msg)
+
+@bfx.wss.once("open")
+async def open():
+    await bfx.wss.subscribe(Channel.STATUS, key="deriv:tBTCF0:USTF0")
+
+bfx.wss.run()

examples/websocket/public/order_book.py

@@ -0,0 +1,65 @@
+# python -c "import examples.websocket.public.order_book"
+
+from collections import OrderedDict
+
+from typing import List
+
+from bfxapi import Client, PUB_WSS_HOST
+
+from bfxapi.websocket import subscriptions
+from bfxapi.websocket.enums import Channel, Error
+from bfxapi.websocket.types import TradingPairBook
+
+class OrderBook(object):
+    def __init__(self, symbols: List[str]):
+        self.__order_book = {
+            symbol: { 
+                "bids": OrderedDict(), "asks": OrderedDict() 
+            } for symbol in symbols
+        }
+            
+    def update(self, symbol: str, data: TradingPairBook) -> None:
+        price, count, amount = data.price, data.count, data.amount
+
+        kind = (amount > 0) and "bids" or "asks"
+
+        if count > 0:
+            self.__order_book[symbol][kind][price] = { 
+                "price": price, 
+                "count": count,
+                "amount": amount 
+            }
+
+        if count == 0:
+            if price in self.__order_book[symbol][kind]:
+                del self.__order_book[symbol][kind][price]
+
+SYMBOLS = [ "tBTCUSD", "tLTCUSD", "tLTCBTC", "tETHUSD", "tETHBTC" ]
+
+order_book = OrderBook(symbols=SYMBOLS)
+
+bfx = Client(WSS_HOST=PUB_WSS_HOST)
+
+@bfx.wss.on("wss-error")
+def on_wss_error(code: Error, msg: str):
+    print(code, msg)
+
+@bfx.wss.on("open")
+async def on_open():
+    for symbol in SYMBOLS:
+        await bfx.wss.subscribe(Channel.BOOK, symbol=symbol)
+
+@bfx.wss.on("subscribed")
+def on_subscribed(subscription):
+    print(f"Subscription successful for pair <{subscription['pair']}>")
+
+@bfx.wss.on("t_book_snapshot")
+def on_t_book_snapshot(subscription: subscriptions.Book, snapshot: List[TradingPairBook]):
+    for data in snapshot:
+        order_book.update(subscription["symbol"], data)
+
+@bfx.wss.on("t_book_update")
+def on_t_book_update(subscription: subscriptions.Book, data: TradingPairBook):
+    order_book.update(subscription["symbol"], data)
+
+bfx.wss.run()

examples/websocket/public/raw_order_book.py

@@ -0,0 +1,65 @@
+# python -c "import examples.websocket.public.raw_order_book"
+
+from collections import OrderedDict
+
+from typing import List
+
+from bfxapi import Client, PUB_WSS_HOST
+
+from bfxapi.websocket import subscriptions
+from bfxapi.websocket.enums import Channel, Error
+from bfxapi.websocket.types import TradingPairRawBook
+
+class RawOrderBook(object):
+    def __init__(self, symbols: List[str]):
+        self.__raw_order_book = {
+            symbol: { 
+                "bids": OrderedDict(), "asks": OrderedDict() 
+            } for symbol in symbols
+        }
+            
+    def update(self, symbol: str, data: TradingPairRawBook) -> None:
+        order_id, price, amount = data.order_id, data.price, data.amount
+
+        kind = (amount > 0) and "bids" or "asks"
+
+        if price > 0:
+            self.__raw_order_book[symbol][kind][order_id] = { 
+                "order_id": order_id,
+                "price": price, 
+                "amount": amount 
+            }
+
+        if price == 0:
+            if order_id in self.__raw_order_book[symbol][kind]:
+                del self.__raw_order_book[symbol][kind][order_id]
+
+SYMBOLS = [ "tBTCUSD", "tLTCUSD", "tLTCBTC", "tETHUSD", "tETHBTC" ]
+
+raw_order_book = RawOrderBook(symbols=SYMBOLS)
+
+bfx = Client(WSS_HOST=PUB_WSS_HOST)
+
+@bfx.wss.on("wss-error")
+def on_wss_error(code: Error, msg: str):
+    print(code, msg)
+
+@bfx.wss.on("open")
+async def on_open():
+    for symbol in SYMBOLS:
+        await bfx.wss.subscribe(Channel.BOOK, symbol=symbol, prec="R0")
+
+@bfx.wss.on("subscribed")
+def on_subscribed(subscription):
+    print(f"Subscription successful for pair <{subscription['pair']}>")
+
+@bfx.wss.on("t_raw_book_snapshot")
+def on_t_raw_book_snapshot(subscription: subscriptions.Book, snapshot: List[TradingPairRawBook]):
+    for data in snapshot:
+        raw_order_book.update(subscription["symbol"], data)
+
+@bfx.wss.on("t_raw_book_update")
+def on_t_raw_book_update(subscription: subscriptions.Book, data: TradingPairRawBook):
+    raw_order_book.update(subscription["symbol"], data)
+
+bfx.wss.run()

examples/websocket/public/ticker.py

@@ -0,0 +1,21 @@
+# python -c "import examples.websocket.public.ticker"
+
+from bfxapi import Client, PUB_WSS_HOST
+
+from bfxapi.websocket import subscriptions
+from bfxapi.websocket.enums import Channel
+from bfxapi.websocket.types import TradingPairTicker
+
+bfx = Client(WSS_HOST=PUB_WSS_HOST)
+
+@bfx.wss.on("t_ticker_update")
+def on_t_ticker_update(subscription: subscriptions.Ticker, data: TradingPairTicker):
+    print(f"Subscription with subId: {subscription['subId']}")
+
+    print(f"Data: {data}")
+
+@bfx.wss.once("open")
+async def open():
+    await bfx.wss.subscribe(Channel.TICKER, symbol="tBTCUSD")
+
+bfx.wss.run()

examples/websocket/public/trades.py

@@ -0,0 +1,29 @@
+# python -c "import examples.websocket.public.trades"
+
+from bfxapi import Client, PUB_WSS_HOST
+from bfxapi.websocket.enums import Error, Channel
+from bfxapi.websocket.types import Candle, TradingPairTrade
+
+from bfxapi.websocket import subscriptions
+
+bfx = Client(WSS_HOST=PUB_WSS_HOST)
+
+@bfx.wss.on("candles_update")
+def on_candles_update(subscription: subscriptions.Candles, candle: Candle):
+  print(f"New candle: {candle}")
+
+@bfx.wss.on("t_trade_executed")
+def on_t_trade_executed(subscription: subscriptions.Trades, trade: TradingPairTrade):
+  print(f"New trade: {trade}")
+
+@bfx.wss.on("wss-error")
+def on_wss_error(code: Error, msg: str):
+    print(code, msg)
+
+@bfx.wss.once("open")
+async def open():
+    await bfx.wss.subscribe(Channel.CANDLES, key="trade:1m:tBTCUSD")
+
+    await bfx.wss.subscribe(Channel.TRADES, symbol="tBTCUSD")
+
+bfx.wss.run()