Moved files into repo

bfxapi/websockets/GenericWebsocket.py

@@ -0,0 +1,58 @@
+import asyncio
+import websockets
+import json
+
+from ..utils.CustomLogger import CustomLogger
+
+class AuthError(Exception): pass
+
+def is_json(myjson):
+  try:
+      json_object = json.loads(myjson)
+  except ValueError as e:
+      return False
+  return True
+
+class GenericWebsocket(object):
+  def __init__(self, host, symbol='tBTCUSD', onCandleHook=None, onTradeHook=None, onCompleteHook=None):
+    if not onCandleHook:
+      raise KeyError("Expected `onCandleHook` in parameters.")
+    if not onTradeHook:
+      raise KeyError("Expected `onTradeHook` in parameters.")
+    if not onCompleteHook:
+      raise KeyError("Expected `onCompleteHook` in parameters.")
+    self.onCandleHook = onCandleHook
+    self.onTradeHook = onTradeHook
+    self.onCompleteHook = onCompleteHook
+    self.symbol = symbol
+    self.host = host
+    self.awaiting_request = False
+
+    self.logger = CustomLogger('HFWebSocket', logLevel='INFO')
+    self.loop = asyncio.get_event_loop()
+    # self.events = EventEmitter()
+
+  def run(self):
+    self.loop.run_until_complete(self._main(self.host))
+
+  async def _main(self, host):
+    async with websockets.connect(host) as websocket:
+      self.ws = websocket
+      while True:
+        await asyncio.sleep(0)
+        message = await websocket.recv()
+        await self.on_message(message)
+
+  async def on_error(self, error):
+    self.logger.error(error)
+
+  async def on_close(self):
+    self.logger.info("Websocket closed.")
+    await self.ws.close()
+    self.onCompleteHook()
+
+  async def on_open(self):
+    pass
+
+  async def on_message(self, message):
+    pass