General fixups

This commit is contained in:
Jacob Plaster
2018-12-14 16:02:33 +00:00
parent c1aea594a8
commit 1cb33692a0
4 changed files with 39 additions and 10 deletions

View File

@@ -26,6 +26,9 @@ class GenericWebsocket(object):
def run(self):
self.loop.run_until_complete(self._main(self.host))
def get_task_executable(self):
return self._main(self.host)
async def _main(self, host):
async with websockets.connect(host) as websocket:
self.ws = websocket
@@ -35,6 +38,9 @@ class GenericWebsocket(object):
message = await websocket.recv()
await self.on_message(message)
def remove_all_listeners(self, event):
self.events.remove_all_listeners(event)
def on(self, event, func=None):
if not func:
return self.events.on(event)