Added support for 'Authorization: Bearer <TOKEN>' header in Request.token property.

Also added a test case for that kind of header.
This commit is contained in:
François KUBLER
2017-06-22 18:11:23 +02:00
parent b5369e611c
commit 55f860da2f
2 changed files with 18 additions and 4 deletions

View File

@@ -86,11 +86,15 @@ class Request(dict):
:return: token related to request
"""
prefixes = ('Token ', 'Bearer ')
auth_header = self.headers.get('Authorization')
if auth_header is not None and 'Token ' in auth_header:
return auth_header.partition('Token ')[-1]
else:
return auth_header
if auth_header is not None:
for prefix in prefixes:
if prefix in auth_header:
return auth_header.partition(prefix)[-1]
return auth_header
@property
def form(self):