Adds SSO auth to gsutil

third_party/gsutil/gslib/util.py

 # Copyright 2010 Google Inc. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
 #
 #     http://www.apache.org/licenses/LICENSE-2.0
 #
 # Unless required by applicable law or agreed to in writing, software
 # distributed under the License is distributed on an "AS IS" BASIS,
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
 """Static data and helper functions."""
 
 import math
 import re
+import os
 import sys
 import time
 
 import boto
 from third_party.retry_decorator.decorators import retry
 
 # We don't use the oauth2 authentication plugin directly; importing it here
 # ensures that it's loaded and available by default. Note: we made this static
 # state instead of Command instance state because the top-level gsutil code
 # needs to check it.
@@ skipping 25 matching lines @@
 
 Retry = retry
 
 # Enum class for specifying listing style.
 class ListingStyle(object):
   SHORT = 'SHORT'
   LONG = 'LONG'
   LONG_LONG = 'LONG_LONG'
 
 
-def HasConfiguredCredentials():
+def HasConfiguredCredentials(bypass_prodaccess):
   """Determines if boto credential/config file exists."""
   config = boto.config
   has_goog_creds = (config.has_option('Credentials', 'gs_access_key_id') and
                     config.has_option('Credentials', 'gs_secret_access_key'))
   has_amzn_creds = (config.has_option('Credentials', 'aws_access_key_id') and
                     config.has_option('Credentials', 'aws_secret_access_key'))
   has_oauth_creds = (HAVE_OAUTH2 and
       config.has_option('Credentials', 'gs_oauth2_refresh_token'))
   has_auth_plugins = config.has_option('Plugin', 'plugin_directory')
+  # Pretend prodaccess doesn't exist if --bypass_prodaccess is passed in.
+  has_prodaccess = HasExecutable('prodaccess') and not bypass_prodaccess
   return (has_goog_creds or has_amzn_creds or has_oauth_creds
-          or has_auth_plugins)
+          or has_auth_plugins or has_prodaccess)
 
 
 def _RoundToNearestExponent(num):
   i = 0
   while i+1 < len(_EXP_STRINGS) and num >= (2 ** _EXP_STRINGS[i+1][0]):
     i += 1
   return i, round(float(num) / 2 ** _EXP_STRINGS[i][0], 2)
 
 def MakeHumanReadable(num):
   """Generates human readable string for a number of bytes.
@@ skipping 60 matching lines @@
     exc_name = str(type(e))
   else:
     exc_name = exc_name_parts[-2]
   if not hasattr(e, 'body'):
     return (exc_name, None)
   detail_start = e.body.find('<Details>')
   detail_end = e.body.find('</Details>')
   if detail_start != -1 and detail_end != -1:
     return (exc_name, e.body[detail_start+9:detail_end])
   return (exc_name, None)
+
+
+def HasExecutable(filename):
+  """Determines if an executable is available on the system."""
+  for path in os.environ['PATH'].split(os.pathsep):
+    exe_file = os.path.join(path, filename)
+    if os.path.exists(exe_file) and os.access(exe_file, os.X_OK):
+      return True
+  return False