Add close method to websocket client

TomasTomecek · Feb 27, 2017 · 94861c7 · 94861c7
1 parent c85b8f5
commit 94861c7
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,6 @@
 # v1.0.1
 - Bugfix: blocking exec call should remove channel metadata #140
+- Add close method to websocket api of interactive exec #145
 
 # v1.0.0b3
 - Bugfix: Missing websocket-client dependency #131

diff --git a/examples/exec.py b/examples/exec.py
@@ -92,3 +92,4 @@
 resp.write_stdin("whoami\n")
 user = resp.readline_stdout(timeout=3)
 print("Server user is: %s" % user)
+resp.close()
diff --git a/kubernetes/client/ws_client.py b/kubernetes/client/ws_client.py
@@ -191,6 +191,14 @@ def run_forever(self, timeout=None):
             while self.is_open():
                 self.update(timeout=None)
 
+    def close(self, **kwargs):
+        """
+        close websocket connection.
+        """
+        self._connected = False
+        if self.sock:
+            self.sock.close(**kwargs)
+
 
 WSResponse = collections.namedtuple('WSResponse', ['data'])