Merge pull request #2240 from lightos/master

stamparm · web-flow · commit 98c6d8f582dc · 2016-10-21T13:03:08.000+02:00
Support for timeout param when using Websockets
diff --git a/lib/request/connect.py b/lib/request/connect.py
@@ -396,6 +396,7 @@ def getPage(**kwargs):
 
             if websocket_:
                 ws = websocket.WebSocket()
+                ws.settimeout(timeout)
                 ws.connect(url, header=("%s: %s" % _ for _ in headers.items() if _[0] not in ("Host",)), cookie=cookie)  # WebSocket will add Host field of headers automatically
                 ws.send(urldecode(post or ""))
                 page = ws.recv()
