inventree/InvenTree
2
0
Fork 0
mirror of https://github.com/inventree/InvenTree.git synced 2025-10-30 20:55:42 +00:00
Code Activity

Token tweaks (#6354)

Browse Source 
- Adjust to allow "bearer" token type
This commit is contained in:
Oliver
2024-01-29 15:58:56 +11:00
committed by GitHub
parent cf7a20e1b7
commit b42f8a620b
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
InvenTree/InvenTree/middleware.py
View File

@@ -21,6 +21,7 @@ logger = logging.getLogger('inventree')
def get_token_from_request(request): def get_token_from_request(request):
"""Extract token information from a request object.""" """Extract token information from a request object."""
auth_keys = ['Authorization', 'authorization'] auth_keys = ['Authorization', 'authorization']
token_keys = ['token', 'bearer']
token = None token = None
@@ -28,9 +29,10 @@ def get_token_from_request(request):
if auth_header := request.headers.get(k, None): if auth_header := request.headers.get(k, None):
auth_header = auth_header.strip().lower().split() auth_header = auth_header.strip().lower().split()
if len(auth_header) > 1 and auth_header[0].startswith('token'): if len(auth_header) > 1:
token = auth_header[1] if auth_header[0].strip().lower().replace(':', '') in token_keys:
break token = auth_header[1]
break
return token return token