Merge pull request #146 from mbohlool/closemethod

Add close method to websocket client

Author: mbohlool

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

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()

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'])