Vasanko/sanic
2
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Merge pull request #811 from Frzk/bearer-support

Browse Source 
Added support for 'Authorization: Bearer <TOKEN>' header...
This commit is contained in:
Eli Uriegas 2017-06-23 10:32:21 -07:00 committed by GitHub
commit dbcbf12456
2 changed files with 18 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
sanic/request.py
View File

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

10
tests/test_requests.py
View File

@ -182,6 +182,16 @@ def test_token():
assert request.token == token assert request.token == token
token = 'a1d895e0-553a-421a-8e22-5ff8ecb48cbf'
headers = {
'content-type': 'application/json',
'Authorization': 'Bearer {}'.format(token)
}
request, response = app.test_client.get('/', headers=headers)
assert request.token == token
# no Authorization headers # no Authorization headers
headers = { headers = {
'content-type': 'application/json' 'content-type': 'application/json'