From 7bd7f060dfceac021b225d9f3e03032e5f2b109b Mon Sep 17 00:00:00 2001 From: Yves Fischer Date: Thu, 5 Jan 2017 18:33:13 +0100 Subject: UI Example --- mp_tool/web.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) (limited to 'mp_tool/web.py') diff --git a/mp_tool/web.py b/mp_tool/web.py index 39e3010..6891aa5 100644 --- a/mp_tool/web.py +++ b/mp_tool/web.py @@ -129,6 +129,8 @@ def eval(url: str, password: str, code: str): ws = _connect_and_auth(url, password) ws.send(Constants.ENTER_REPL_MODE) stdout.write(read_until_eval_or_timeout(ws)) + + ws.settimeout(5) ws.send(code + "\r\n") result = read_until_eval_or_timeout(ws) @@ -139,8 +141,11 @@ def eval(url: str, password: str, code: str): def read_until_eval_or_timeout(ws: websocket.WebSocket): buf = "" - while not buf.endswith("\r\n>>> "): - buf += ws.recv() + try: + while not buf.endswith("\r\n>>> "): + buf += ws.recv() + except websocket.WebSocketTimeoutException: + pass return buf -- cgit v1.2.1