mirror of
https://github.com/aljazceru/bitfinex-api-py.git
synced 2025-12-19 14:54:21 +01:00
63 lines
1.6 KiB
Python
63 lines
1.6 KiB
Python
import asyncio
|
|
import websockets
|
|
import json
|
|
|
|
from pyee import EventEmitter
|
|
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, logLevel='ERROR'):
|
|
self.host = host
|
|
self.logger = CustomLogger('HFWebSocket', logLevel=logLevel)
|
|
self.loop = asyncio.get_event_loop()
|
|
self.events = EventEmitter(scheduler=asyncio.ensure_future, loop=self.loop)
|
|
|
|
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)
|
|
|
|
def on(self, event, func=None):
|
|
if not func:
|
|
return self.events.on(event)
|
|
self.events.on(event, func)
|
|
|
|
def once(self, event, func=None):
|
|
if not func:
|
|
return self.events.once(event)
|
|
self.events.once(event, func)
|
|
|
|
def _emit(self, event, *args, **kwargs):
|
|
self.events.emit(event, *args, **kwargs)
|
|
|
|
async def on_error(self, error):
|
|
self.logger.error(error)
|
|
self.events.emit('error', error)
|
|
|
|
async def on_close(self):
|
|
self.logger.info("Websocket closed.")
|
|
await self.ws.close()
|
|
self._emit('done')
|
|
|
|
async def on_open(self):
|
|
pass
|
|
|
|
async def on_message(self, message):
|
|
pass
|