Check if token is empty or invalid, update postman

webservice/helper_functions.py

@@ -3,7 +3,7 @@ import os
 import uuid
 
 import jwt
-from flask import request
+from flask import request, jsonify
 
 from db import db
 from models import User
@@ -33,11 +33,23 @@ def extract_token_data(token):
             return jwt.decode(token, os.getenv('SECRET_KEY'), algorithms=["HS256"])
         except:
             return None
+    else:
+        return None
 
 
 def get_username_from_token_data(token_data):
-    return token_data['username']
+    if token_data is not None:
+        return token_data['username']
+    else:
+        return None
 
 
 def get_user_id_from_username(username):
-    return db.session.query(User).filter_by(username=username).first().user_id
+    if username is not None:
+        return db.session.query(User).filter_by(username=username).first().user_id
+    else:
+        return None
+
+
+def return_401():
+    return jsonify({"status": 401, "text": "Authorization token not provided or not valid"})