mirror of
https://github.com/jointakahe/takahe.git
synced 2024-11-29 10:40:59 +00:00
28 lines
905 B
Python
28 lines
905 B
Python
|
from django.http import HttpResponse
|
||
|
|
|
||
|
|
from api.models import Token
|
||
|
|
|
||
|
|
|
||
|
|
class ApiTokenMiddleware:
|
||
|
|
"""
|
||
|
|
Adds request.user and request.identity if an API token appears.
|
||
|
|
Also nukes request.session so it can't be used accidentally.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def __init__(self, get_response):
|
||
|
|
self.get_response = get_response
|
||
|
|
|
||
|
|
def __call__(self, request):
|
||
|
|
auth_header = request.headers.get("authorization", None)
|
||
|
|
if auth_header and auth_header.startswith("Bearer "):
|
||
|
|
token_value = auth_header[7:]
|
||
|
|
try:
|
||
|
|
token = Token.objects.get(token=token_value)
|
||
|
|
except Token.DoesNotExist:
|
||
|
|
return HttpResponse("Invalid Bearer token", status=400)
|
||
|
|
request.user = token.user
|
||
|
|
request.identity = token.identity
|
||
|
|
request.session = None
|
||
|
|
response = self.get_response(request)
|
||
|
|
return response
|