mirror of
https://github.com/aljazceru/bitfinex-api-py.git
synced 2025-12-19 23:04:21 +01:00
fix all linting for pylint3
This commit is contained in:
@@ -1,3 +1,7 @@
|
||||
"""
|
||||
Module used as a interfeace to describe a generick websocket client
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import websockets
|
||||
import json
|
||||
@@ -5,66 +9,106 @@ import json
|
||||
from pyee import EventEmitter
|
||||
from ..utils.CustomLogger import CustomLogger
|
||||
|
||||
class AuthError(Exception): pass
|
||||
|
||||
class AuthError(Exception):
|
||||
"""
|
||||
Thrown whenever there is a problem with the authentication packet
|
||||
"""
|
||||
pass
|
||||
|
||||
|
||||
def is_json(myjson):
|
||||
try:
|
||||
json_object = json.loads(myjson)
|
||||
except ValueError as e:
|
||||
return False
|
||||
return True
|
||||
try:
|
||||
json_object = json.loads(myjson)
|
||||
except ValueError as e:
|
||||
return False
|
||||
return True
|
||||
|
||||
class GenericWebsocket(object):
|
||||
|
||||
def __init__(self, host, logLevel='INFO', loop=None):
|
||||
self.host = host
|
||||
self.logger = CustomLogger('BfxWebsocket', logLevel=logLevel)
|
||||
self.loop = loop or asyncio.get_event_loop()
|
||||
self.events = EventEmitter(scheduler=asyncio.ensure_future, loop=self.loop)
|
||||
self.ws = None
|
||||
class GenericWebsocket:
|
||||
"""
|
||||
Websocket object used to contain the base functionality of a websocket.
|
||||
Inlcudes an event emitter and a standard websocket client.
|
||||
"""
|
||||
|
||||
def run(self):
|
||||
self.loop.run_until_complete(self._main(self.host))
|
||||
def __init__(self, host, logLevel='INFO', loop=None):
|
||||
self.host = host
|
||||
self.logger = CustomLogger('BfxWebsocket', logLevel=logLevel)
|
||||
self.loop = loop or asyncio.get_event_loop()
|
||||
self.events = EventEmitter(
|
||||
scheduler=asyncio.ensure_future, loop=self.loop)
|
||||
self.ws = None
|
||||
|
||||
def get_task_executable(self):
|
||||
return self._main(self.host)
|
||||
def run(self):
|
||||
"""
|
||||
Run the websocket connection indefinitely
|
||||
"""
|
||||
self.loop.run_until_complete(self._main(self.host))
|
||||
|
||||
async def _main(self, host):
|
||||
async with websockets.connect(host) as websocket:
|
||||
self.ws = websocket
|
||||
self.logger.info("Wesocket connectedt to {}".format(self.host))
|
||||
while True:
|
||||
await asyncio.sleep(0)
|
||||
message = await websocket.recv()
|
||||
await self.on_message(message)
|
||||
def get_task_executable(self):
|
||||
"""
|
||||
Get the run indefinitely asyncio task
|
||||
"""
|
||||
return self._main(self.host)
|
||||
|
||||
def remove_all_listeners(self, event):
|
||||
self.events.remove_all_listeners(event)
|
||||
async def _main(self, host):
|
||||
async with websockets.connect(host) as websocket:
|
||||
self.ws = websocket
|
||||
self.logger.info("Wesocket connectedt to {}".format(self.host))
|
||||
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 remove_all_listeners(self, event):
|
||||
"""
|
||||
Remove all listeners from event emitter
|
||||
"""
|
||||
self.events.remove_all_listeners(event)
|
||||
|
||||
def once(self, event, func=None):
|
||||
if not func:
|
||||
return self.events.once(event)
|
||||
self.events.once(event, func)
|
||||
def on(self, event, func=None):
|
||||
"""
|
||||
Add a new event to the event emitter
|
||||
"""
|
||||
if not func:
|
||||
return self.events.on(event)
|
||||
self.events.on(event, func)
|
||||
|
||||
def _emit(self, event, *args, **kwargs):
|
||||
self.events.emit(event, *args, **kwargs)
|
||||
def once(self, event, func=None):
|
||||
"""
|
||||
Add a new event to only fire once to the event
|
||||
emitter
|
||||
"""
|
||||
if not func:
|
||||
return self.events.once(event)
|
||||
self.events.once(event, func)
|
||||
|
||||
async def on_error(self, error):
|
||||
self.logger.error(error)
|
||||
self.events.emit('error', error)
|
||||
def _emit(self, event, *args, **kwargs):
|
||||
self.events.emit(event, *args, **kwargs)
|
||||
|
||||
async def on_close(self):
|
||||
self.logger.info("Websocket closed.")
|
||||
await self.ws.close()
|
||||
self._emit('done')
|
||||
async def on_error(self, error):
|
||||
"""
|
||||
On websocket error print and fire event
|
||||
"""
|
||||
self.logger.error(error)
|
||||
self.events.emit('error', error)
|
||||
|
||||
async def on_open(self):
|
||||
pass
|
||||
async def on_close(self):
|
||||
"""
|
||||
On websocket close print and fire event
|
||||
"""
|
||||
self.logger.info("Websocket closed.")
|
||||
await self.ws.close()
|
||||
self._emit('done')
|
||||
|
||||
async def on_message(self, message):
|
||||
pass
|
||||
async def on_open(self):
|
||||
"""
|
||||
On websocket open
|
||||
"""
|
||||
pass
|
||||
|
||||
async def on_message(self, message):
|
||||
"""
|
||||
On websocket message
|
||||
"""
|
||||
pass
|
||||
|
||||
Reference in New Issue
Block a user