Update ws_client.py to support proxy

kubernetes-client · Aug 29, 2019 · 34f3d05 · 34f3d05
1 parent 4b8e89f
commit 34f3d05
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/stream/ws_client.py b/stream/ws_client.py
@@ -74,7 +74,11 @@ def __init__(self, configuration, url, headers):
             ssl_opts['keyfile'] = configuration.key_file
 
         self.sock = WebSocket(sslopt=ssl_opts, skip_utf8_validation=False)
-        self.sock.connect(url, header=header)
+        if configuration.proxy:
+            proxy_url = urlparse(configuration.proxy)
+            self.sock.connect(url, header=header, http_proxy_host=proxy_url.hostname, http_proxy_port=proxy_url.port)
+        else:
+            self.sock.connect(url, header=header)
         self._connected = True
 
     def peek_channel(self, channel, timeout=0):