diff options
author | Devaev Maxim <[email protected]> | 2018-11-06 01:55:13 +0300 |
---|---|---|
committer | Devaev Maxim <[email protected]> | 2018-11-06 01:55:13 +0300 |
commit | f0ae427d8e4bbd82653abdb8e2aef8ffe32fc732 (patch) | |
tree | 44192d6bce4cc80db95532df2888e0764f63d58d /kvmd/extras/wscli | |
parent | 1ac968e92411409413ed4555eed6c2944b38de0d (diff) |
refactoring
Diffstat (limited to 'kvmd/extras/wscli')
-rw-r--r-- | kvmd/extras/wscli/__init__.py | 50 | ||||
-rw-r--r-- | kvmd/extras/wscli/__main__.py | 2 |
2 files changed, 0 insertions, 52 deletions
diff --git a/kvmd/extras/wscli/__init__.py b/kvmd/extras/wscli/__init__.py deleted file mode 100644 index 19d3181b..00000000 --- a/kvmd/extras/wscli/__init__.py +++ /dev/null @@ -1,50 +0,0 @@ -#!/usr/bin/env python3 - - -import sys -import signal -import asyncio -import argparse -import time - -import aiohttp - - -# ===== -async def _run_client(loop: asyncio.AbstractEventLoop, url: str) -> None: - def stdin_callback() -> None: - line = sys.stdin.buffer.readline().decode() - if line: - asyncio.ensure_future(ws.send_str(line), loop=loop) - else: - loop.stop() - - loop.add_reader(sys.stdin.fileno(), stdin_callback) - - async def dispatch() -> None: - while True: - msg = await ws.receive() - if msg.type == aiohttp.WSMsgType.TEXT: - print("[%.5f] Received: %s" % (time.time(), msg.data.strip())) - else: - if msg.type == aiohttp.WSMsgType.CLOSE: - await ws.close() - elif msg.type == aiohttp.WSMsgType.ERROR: - print("[%.5f] Error during receive: %s" % (time.time(), ws.exception())) - elif msg.type == aiohttp.WSMsgType.CLOSED: - pass - break - - async with aiohttp.ClientSession().ws_connect(url) as ws: - await dispatch() - - -def main() -> None: - parser = argparse.ArgumentParser() - parser.add_argument("-u", "--url", default="http://127.0.0.1:8081/ws") - options = parser.parse_args() - - loop = asyncio.get_event_loop() - loop.add_signal_handler(signal.SIGINT, loop.stop) - loop.create_task(_run_client(loop, options.url)) - loop.run_forever() diff --git a/kvmd/extras/wscli/__main__.py b/kvmd/extras/wscli/__main__.py deleted file mode 100644 index 031df43e..00000000 --- a/kvmd/extras/wscli/__main__.py +++ /dev/null @@ -1,2 +0,0 @@ -from . import main -main() |