Using basic auth for proxy tunnel

Author: Pascal Zingg

Diff for: plotly/plotly/chunked_requests/chunked_request.py

@@ -4,7 +4,9 @@
 import ssl
 
 from six.moves import http_client
-from six.moves.urllib.parse import urlparse
+from six.moves.urllib.parse import urlparse, unquote
+
+from plotly.api import utils
 
 
 class Stream:
@@ -86,6 +88,7 @@ def _get_proxy_config(self):
 
         proxy_server = None
         proxy_port = None
+        proxy_auth = None
         ssl_enabled = self._ssl_enabled
 
         if ssl_enabled:
@@ -100,7 +103,12 @@ def _get_proxy_config(self):
             proxy_server = p.hostname
             proxy_port = p.port
 
-        return proxy_server, proxy_port
+        if p.username and p.password:
+            username = unquote(p.username)
+            password = unquote(p.password)
+            proxy_auth = utils.basic_auth(username, password)
+
+        return proxy_server, proxy_port, proxy_auth
 
     def _get_ssl_context(self):
         """
@@ -123,7 +131,7 @@ def _connect(self):
         port = self._port
         headers = self._headers
         ssl_enabled = self._ssl_enabled
-        proxy_server, proxy_port = self._get_proxy_config()
+        proxy_server, proxy_port, proxy_auth = self._get_proxy_config()
 
         if (proxy_server and proxy_port):
             if ssl_enabled:
@@ -135,7 +143,12 @@ def _connect(self):
                 self._conn = http_client.HTTPConnection(
                     proxy_server, proxy_port
                 )
-            self._conn.set_tunnel(server, port)
+
+            tunnel_headers = None
+            if proxy_auth:
+                tunnel_headers = {'Proxy-Authorization': proxy_auth}
+
+            self._conn.set_tunnel(server, port, headers=tunnel_headers)
         else:
             if ssl_enabled:
                 context = self._get_ssl_context()