Work in progress: bitbucket support

util/oauth.py

@@ -33,6 +33,36 @@ class OAuthConfig(object):
 
     return endpoint
 
+  def exchange_code_for_token(self, app_config, http_client, code, form_encode=False,
+                              redirect_suffix=''):
+    payload = {
+      'client_id': self.client_id(),
+      'client_secret': self.client_secret(),
+      'code': code,
+      'grant_type': 'authorization_code',
+      'redirect_uri': '%s://%s/oauth2/%s/callback%s' % (app_config['PREFERRED_URL_SCHEME'],
+                                                        app_config['SERVER_HOSTNAME'],
+                                                        self.service_name().lower(),
+                                                        redirect_suffix)
+    }
+
+    headers = {
+      'Accept': 'application/json'
+    }
+
+    token_url = self.token_endpoint()
+    if form_encode:
+      get_access_token = http_client.post(token_url, data=payload, headers=headers)
+    else:
+      get_access_token = http_client.post(token_url, params=payload, headers=headers)
+
+    json_data = get_access_token.json()
+    if not json_data:
+      return ''
+
+    token = json_data.get('access_token', '')
+    return token
+
 
 class GithubOAuthConfig(OAuthConfig):
   def __init__(self, config, key_name):