updated the token validator

mmohades · Oct 23, 2020 · 7862739 · 7862739
1 parent a3858c0
commit 7862739
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/venmo_api/utils/api_util.py b/venmo_api/utils/api_util.py
@@ -1,6 +1,7 @@
 from enum import Enum
 from typing import Dict, List
 from venmo_api import ArgumentMissingError, User
+import re
 
 
 def validate_access_token(access_token):
@@ -9,13 +10,13 @@ def validate_access_token(access_token):
     :param access_token:
     :return:
     """
+    token_re = r'^(Bearer)?(.+)$'
     if not access_token:
         return
 
-    if access_token[:6] != 'Bearer':
-        return f"Bearer {access_token}"
+    access_token = re.findall(token_re, access_token)[0][1].replace(' ', '')
 
-    return access_token
+    return f"Bearer {access_token}"
 
 
 def deserialize(response: Dict, data_type, nested_response: List[str] = None):

diff --git a/venmo_api/venmo.py b/venmo_api/venmo.py
@@ -46,4 +46,5 @@ def log_out(access_token) -> bool:
         :param access_token:
         :return: <bool>
         """
+        access_token = validate_access_token(access_token=access_token)
         return AuthenticationApi.log_out(access_token=access_token)