adding the option to set Cookie as a header name (#29)

jetbridge · Apr 13, 2024 · 76f629c · 76f629c
1 parent 3f5d0e6
commit 76f629c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/flask_cognito.py b/flask_cognito.py
@@ -88,7 +88,10 @@ def get_token(self):
         auth_header_prefix = _cog.jwt_header_prefix
 
         # get token value from header
-        auth_header_value = request.headers.get(auth_header_name)
+        if auth_header_name == 'Cookie':
+            return request.cookies.get('session_token')
+        else:
+            auth_header_value = request.headers.get(auth_header_name)
 
         if not auth_header_value:
             # no auth header found