def testCacheFileName(self):
"""Tests configuring the cache with a specific file name."""
cache = oauth2_client.FileSystemTokenCache(
path_pattern='/var/run/ccache/token.%(uid)s.%(key)s')
if IS_WINDOWS:
uid = '_'
else:
uid = os.getuid()
self.assertEqual('/var/run/ccache/token.%s.abc123' % uid,
cache.CacheFileName('abc123'))
cache = oauth2_client.FileSystemTokenCache(
path_pattern='/var/run/ccache/token.%(key)s')
self.assertEqual('/var/run/ccache/token.abc123',
cache.CacheFileName('abc123'))
def setUp(self):
self.cache = oauth2_client.FileSystemTokenCache()
self.start_time = datetime.datetime(2011, 3, 1, 10, 25, 13, 300826)
self.token_1 = oauth2_client.AccessToken('token1', self.start_time)
self.token_2 = oauth2_client.AccessToken(
'token2', self.start_time + datetime.timedelta(seconds=492))
self.key = 'token1key'
def OAuth2ClientFromBotoConfig(
config, cred_type=oauth2_client.CredTypes.OAUTH2_USER_ACCOUNT):
token_cache = None
token_cache_type = config.get('OAuth2', 'token_cache', 'file_system')
if token_cache_type == 'file_system':
if config.has_option('OAuth2', 'token_cache_path_pattern'):
token_cache = oauth2_client.FileSystemTokenCache(
path_pattern=config.get('OAuth2', 'token_cache_path_pattern'))
else:
token_cache = oauth2_client.FileSystemTokenCache()
elif token_cache_type == 'in_memory':
token_cache = oauth2_client.InMemoryTokenCache()
else:
raise Exception(
"Invalid value for config option OAuth2/token_cache: %s" %
token_cache_type)
proxy_host = None
proxy_port = None
proxy_user = None
proxy_pass = None
if (config.has_option('Boto', 'proxy')
and config.has_option('Boto', 'proxy_port')):
proxy_host = config.get('Boto', 'proxy')
proxy_port = int(config.get('Boto', 'proxy_port'))
proxy_user = config.get('Boto', 'proxy_user', None)
proxy_pass = config.get('Boto', 'proxy_pass', None)
provider_authorization_uri = config.get(
'OAuth2', 'provider_authorization_uri',
GOOGLE_OAUTH2_PROVIDER_AUTHORIZATION_URI)
provider_token_uri = config.get(
'OAuth2', 'provider_token_uri', GOOGLE_OAUTH2_PROVIDER_TOKEN_URI)
if cred_type == oauth2_client.CredTypes.OAUTH2_SERVICE_ACCOUNT:
service_client_id = config.get('Credentials', 'gs_service_client_id', '')
private_key_filename = config.get('Credentials', 'gs_service_key_file', '')
with open(private_key_filename, 'rb') as private_key_file:
private_key = private_key_file.read()
json_key = None
try:
json_key = json.loads(private_key)
except ValueError:
pass
if json_key:
for json_entry in ('client_id', 'client_email', 'private_key_id',
'private_key'):
if json_entry not in json_key:
raise Exception('The JSON private key file at %s '
'did not contain the required entry: %s' %
(private_key_filename, json_entry))
return oauth2_client.OAuth2JsonServiceAccountClient(
json_key['client_id'], json_key['client_email'],
json_key['private_key_id'], json_key['private_key'],
access_token_cache=token_cache, auth_uri=provider_authorization_uri,
token_uri=provider_token_uri,
disable_ssl_certificate_validation=not(config.getbool(
'Boto', 'https_validate_certificates', True)),
proxy_host=proxy_host, proxy_port=proxy_port,
proxy_user=proxy_user, proxy_pass=proxy_pass)
else:
key_file_pass = config.get('Credentials', 'gs_service_key_file_password',
GOOGLE_OAUTH2_DEFAULT_FILE_PASSWORD)
return oauth2_client.OAuth2ServiceAccountClient(
service_client_id, private_key, key_file_pass,
access_token_cache=token_cache, auth_uri=provider_authorization_uri,
token_uri=provider_token_uri,
disable_ssl_certificate_validation=not(config.getbool(
'Boto', 'https_validate_certificates', True)),
proxy_host=proxy_host, proxy_port=proxy_port,
proxy_user=proxy_user, proxy_pass=proxy_pass)
elif cred_type == oauth2_client.CredTypes.OAUTH2_USER_ACCOUNT:
client_id = config.get('OAuth2', 'client_id',
os.environ.get('OAUTH2_CLIENT_ID', CLIENT_ID))
if not client_id:
raise Exception(
'client_id for your application obtained from '
'https://console.developers.google.com must be set in a boto config '
'or with OAUTH2_CLIENT_ID environment variable or with '
'gcs_oauth2_boto_plugin.SetFallbackClientIdAndSecret function.')
client_secret = config.get('OAuth2', 'client_secret',
os.environ.get('OAUTH2_CLIENT_SECRET',
CLIENT_SECRET))
ca_certs_file=config.get_value('Boto', 'ca_certificates_file')
if ca_certs_file == 'system':
ca_certs_file = None
if not client_secret:
raise Exception(
'client_secret for your application obtained from '
'https://console.developers.google.com must be set in a boto config '
'or with OAUTH2_CLIENT_SECRET environment variable or with '
'gcs_oauth2_boto_plugin.SetFallbackClientIdAndSecret function.')
return oauth2_client.OAuth2UserAccountClient(
provider_token_uri, client_id, client_secret,
config.get('Credentials', 'gs_oauth2_refresh_token'),
auth_uri=provider_authorization_uri, access_token_cache=token_cache,
disable_ssl_certificate_validation=not(config.getbool(
'Boto', 'https_validate_certificates', True)),
proxy_host=proxy_host, proxy_port=proxy_port,
proxy_user=proxy_user, proxy_pass=proxy_pass,
ca_certs_file=ca_certs_file)
else:
raise Exception('You have attempted to create an OAuth2 client without '
'setting up OAuth2 credentials.')