modifying oauth2 flow to check for the team_id key and populate the resulting object accordingly

Alex Swehla · posita · commit 299532de5763 · 2017-07-18T10:50:45.000-07:00
diff --git a/dropbox/oauth.py b/dropbox/oauth.py
@@ -120,11 +120,19 @@ def _finish(self, code, redirect_uri):
         resp.raise_for_status()
 
         d = resp.json()
+        
+        access_token = d['access_token']
+        uid = d['uid']
+        if 'team_id' in d:
+            account_id = d['team_id']
+        else:
+            account_id = d['account_id']
+
 
         return OAuth2FlowNoRedirectResult(
-            d['access_token'],
-            d['account_id'],
-            d['uid'])
+            access_token,
+            account_id,
+            uid)
 
     def build_path(self, target, params=None):
         """Build the path component for an API URL.
