Index: gsutil.py |
diff --git a/gsutil.py b/gsutil.py |
index 6578d7844171202eb787bf1414c4a371e15a75a0..bdb9d53d7b3f939c8a943533dabba1025b2e6863 100755 |
--- a/gsutil.py |
+++ b/gsutil.py |
@@ -15,6 +15,7 @@ import shutil |
import subprocess |
import sys |
import urllib2 |
+import urlparse |
import zipfile |
@@ -121,12 +122,32 @@ def ensure_gsutil(version, target): |
return gsutil_bin |
+def get_boto_proxy_options_from_environment(): |
Ryan Tseng
2015/01/07 22:21:58
for completeness sake (since I've seen all these p
|
+ """Converts the contents of the 'http_proxy' environment variable into the |
Ryan Tseng
2015/01/07 22:21:58
"""One line
More lines
...
"""
So rewrite this a
|
+ corresponding Boto options that gsutil accepts.""" |
+ if 'http_proxy' not in os.environ: |
+ return [] |
+ proxy_url = urlparse.urlsplit(os.environ['http_proxy']) |
+ if proxy_url.hostname is None or proxy_url.port is None: |
+ raise EnvironmentError( |
+ 'The "http_proxy" environment is set, but has an invalid format. Its ' |
Ryan Tseng
2015/01/07 22:21:58
2 indents
|
+ 'format must be "http://[user[:password]]host:port".') |
+ boto_options = ['-o', 'Boto:proxy=%s' % proxy_url.hostname, |
+ '-o', 'Boto:proxy_port=%d' % proxy_url.port] |
+ if proxy_url.username: |
+ boto_options.extend(['-o', 'Boto:proxy_user=%s' % proxy_url.username]) |
+ if proxy_url.password: |
+ boto_options.extend(['-o', 'Boto:proxy_pass=%s' % proxy_url.password]) |
+ return boto_options |
+ |
+ |
def run_gsutil(force_version, fallback, target, args): |
if force_version: |
gsutil_bin = ensure_gsutil(force_version, target) |
else: |
gsutil_bin = fallback |
- cmd = [sys.executable, gsutil_bin] + args |
+ proxy_options = get_boto_proxy_options_from_environment() |
+ cmd = [sys.executable, gsutil_bin] + proxy_options + args |
try: |
call(cmd) |
except SubprocessError as e: |