You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
25 lines
774 B
Python
25 lines
774 B
Python
12 years ago
|
from boto.auth_handler import AuthHandler
|
||
|
|
from boto.auth_handler import NotReadyToAuthenticate
|
||
|
|
import oauth2_client
|
||
|
|
import oauth2_helper
|
||
|
|
|
||
|
|
class OAuth2Auth(AuthHandler):
|
||
|
|
|
||
|
|
capability = ['google-oauth2', 's3']
|
||
|
|
|
||
|
|
def __init__(self, path, config, provider):
|
||
|
|
if (provider.name == 'google'
|
||
|
|
and config.has_option('Credentials', 'gs_oauth2_refresh_token')):
|
||
|
|
|
||
|
|
self.oauth2_client = oauth2_helper.OAuth2ClientFromBotoConfig(config)
|
||
|
|
|
||
|
|
self.refresh_token = oauth2_client.RefreshToken(
|
||
|
|
self.oauth2_client,
|
||
|
|
config.get('Credentials', 'gs_oauth2_refresh_token'))
|
||
|
|
else:
|
||
|
|
raise NotReadyToAuthenticate()
|
||
|
|
|
||
|
|
def add_auth(self, http_request):
|
||
|
|
http_request.headers['Authorization'] = \
|
||
|
|
self.refresh_token.GetAuthorizationHeader()
|