more fancy error catching in c2stuff

2017-02-25 19:50:51 +01:00
parent 60ffe17d6b
commit a4b98ed852
1 changed files with 6 additions and 1 deletions
--- a/rowers/c2stuff.py
+++ b/rowers/c2stuff.py
@@ -351,7 +351,12 @@ def get_token(code):

    token_json = response.json()

-    if token_json['status_code'] == 200:
+    try:
+        status_code = token_json['status_code']
+    except KeyError:
+        status_code = token_json.status_code
+
+    if status_code == 200:
        thetoken = token_json['access_token']
        expires_in = token_json['expires_in']
        refresh_token = token_json['refresh_token']