From 327896abccd1c448c37f9007b6a995ba477a9ac8 Mon Sep 17 00:00:00 2001
From: Swapnil <swapnil.theskss@gmail.com>
Date: Tue, 30 Jan 2024 21:10:04 +0530
Subject: [PATCH] app: fixed bug and better error message

---
 app/main.py | 22 ++++++++++++++--------
 1 file changed, 14 insertions(+), 8 deletions(-)

diff --git a/app/main.py b/app/main.py
index eff76a9..4671f65 100644
--- a/app/main.py
+++ b/app/main.py
@@ -28,7 +28,7 @@ def home():
     if is_logged_in():
         token_id = session.get('token_id')
         resp = requests.get(f'{database_url}/token/{token_id}')
-        resp.raise_for_status()
+        raise_for_status(resp)
         resp_json = resp.json()
         token = resp_json['token']
 
@@ -89,7 +89,7 @@ def oauth_redirect():
         }
     )
 
-    response.raise_for_status()
+    raise_for_status(response)
 
     token = response.json()
 
@@ -97,9 +97,13 @@ def oauth_redirect():
     token['refresh_token'] = 'N/A'
     token['expires_in'] = 3600
 
+    user_info = requests.get('https://api.github.com/user', headers={
+        'Authorization': f'Bearer {token.get("access_token")}'
+        }).json()
+
     # Save tokens for later use
     token_id = save_token(
-        token.get('email'), # for inoreader it's userEmail
+        user_info.get('email'), # for inoreader it's userEmail
         token.get('access_token'),
         token.get('refresh_token'),
         token.get('expires_in')
@@ -140,7 +144,7 @@ def submit_readwise_api():
     }, json={
         'readwise_api_key': request.form.get('readwise_api_key')
     })
-    response.raise_for_status()
+    raise_for_status(response)
 
     return redirect(url_for('home'))
 
@@ -150,7 +154,7 @@ def is_logged_in():
         return False
 
     response = requests.get(f'{database_url}/token/{token_id}')
-    response.raise_for_status()
+    raise_for_status(response)
     resp_json = response.json()
     token = resp_json['token']
 
@@ -169,14 +173,16 @@ def save_token(email, access_token, refresh_token, expiration_seconds):
             'expiration_seconds': expiration_seconds
         }
     )
-    # response.raise_for_status()
-    if response.status_code not in range(200, 300):
-        raise Exception(f'HTTPError: {response.status_code} \n Message: {response.json()}')
+    raise_for_status(response)
 
     return response.json().get('id')
 
 def set_session_token_id(token_id):
     session['token_id'] = token_id
 
+def raise_for_status(response):
+    if response.status_code not in range(200, 300):
+        raise Exception(f'HTTPError: {response.status_code} \n Message: {response.json()}')
+
 if __name__ == '__main__':
     app.run(host='0.0.0.0', debug=True, port=5000)
\ No newline at end of file