Merge pull request #26 from WebEngineering2/try_except_fix

Try except fix

api/auth.py

@@ -17,5 +17,5 @@ def verify_token(token):
     try:
         jwt.decode(token, os.getenv('SECRET_KEY'), algorithms=["HS256"])
         return True
-    except:
+    except jwt.PyJWTError:
         return False

api/helper_functions.py

@@ -32,7 +32,7 @@ def extract_token_data(token):
     if token is not None:
         try:
             return jwt.decode(token, os.getenv('SECRET_KEY'), algorithms=["HS256"])
-        except:
+        except jwt.PyJWTError:
             return None
     else:
         return None
@@ -52,13 +52,13 @@ def get_email_from_token_data():
                         return email
                     else:
                         return None
-                except:
+                except jwt.PyJWTError:
                     return None
 
             else:  # "Normal" token, extract username from token
                 try:
                     return jwt.decode(token, os.getenv('SECRET_KEY'), algorithms=["HS256"])['email']
-                except:
+                except jwt.PyJWTError:
                     return None
 
     return None

api/schema.py

@@ -60,16 +60,31 @@ class KeywordSchema(Schema):
     keyword = String()
 
 
+class SymbolSchema(Schema):
+    symbol = String()
+
+
+class TransactionSchema(Schema):
+    symbol = String()
+    time = String(validate=validate.Regexp(r"\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}\.\d{3}Z"))
+    count = Integer()
+    price = Float()
+
+
+class TelegramIdSchema(Schema):
+    telegram_user_id = String()
+
+
+class DeleteSuccessfulSchema(Schema):
+    pass
+
+
 class KeywordResponseSchema(Schema):
     keyword = String()
     s_id = Integer()
     email = Email()
 
 
-class SymbolSchema(Schema):
-    symbol = String()
-
-
 class SymbolResponseSchema(Schema):
     symbol = String()
     s_id = Integer()
@@ -81,13 +96,6 @@ class PortfolioShareResponseSchema(Schema):
     last_transaction = String()
 
 
-class TransactionSchema(Schema):
-    symbol = String()
-    time = String(validate=validate.Regexp(r"\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}\.\d{3}Z"))
-    count = Integer()
-    price = Float()
-
-
 class TransactionResponseSchema(Schema):
     email = Email()
     symbol = String()
@@ -96,16 +104,8 @@ class TransactionResponseSchema(Schema):
     price = Float()
 
 
-class TelegramIdSchema(Schema):
-    telegram_user_id = String()
-
-
 class PortfolioResponseSchema(Schema):
     symbol = String()
     last_transaction = String()
     count = Integer()
     # price = Float()
-
-
-class DeleteSuccessfulSchema(Schema):
-    pass