kubernetes-client/python
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add proxy authentication supporting for websocket (stream/ws_client.py)

Browse Source
This commit is contained in:
itaru2622 2021-09-29 09:18:55 +09:00
parent b0afc93ffa
commit 769bc57ec7
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
stream/ws_client.py
View File

@ -29,6 +29,7 @@ from six.moves.urllib.parse import urlencode, urlparse, urlunparse
from six import StringIO
from websocket import WebSocket, ABNF, enableTrace
from base64 import b64decode
STDIN_CHANNEL = 0
STDOUT_CHANNEL = 1
@ -445,11 +446,20 @@ def create_websocket(configuration, url, headers=None):
ssl_opts['keyfile'] = configuration.key_file
websocket = WebSocket(sslopt=ssl_opts, skip_utf8_validation=False)
connect_opt = {
'header': header
}
if configuration.proxy:
proxy_url = urlparse(configuration.proxy)
websocket.connect(url, header=header, http_proxy_host=proxy_url.hostname, http_proxy_port=proxy_url.port)
else:
websocket.connect(url, header=header)
connect_opt.update({'http_proxy_host': proxy_url.hostname, 'http_proxy_port': proxy_url.port})
if configuration.proxy_headers:
for key,value in configuration.proxy_headers.items():
if key == 'proxy-authorization' and value.startswith('Basic'):
b64value = value.split()[1]
auth = b64decode(b64value).decode().split(':')
connect_opt.update({'http_proxy_auth': (auth[0], auth[1]) })
websocket.connect(url, **connect_opt)
return websocket