Merge pull request #55 from nunofgs/bugfix/add-timeout-support-for-we…

…bsocket-connections Add timeout support for websocket connections
Ape · Oct 31, 2017 · 0b42a0b · 0b42a0b
2 parents 86130ea + fd1606b
commit 0b42a0b
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/samsungctl/remote_websocket.py b/samsungctl/remote_websocket.py
@@ -18,9 +18,8 @@ def __init__(self, config):
         URL_FORMAT = "ws://{}:{}/api/v2/channels/samsung.remote.control?name={}"
 
         """Make a new connection."""
-        self.connection = websocket.WebSocket()
-        self.connection.connect(URL_FORMAT.format(config["host"], config["port"],
-                                                  self._serialize_string(config["name"])))
+        self.connection = websocket.create_connection(URL_FORMAT.format(config["host"], config["port"],
+                                                  self._serialize_string(config["name"])), config["timeout"])
 
         self._read_response()