mirror of
https://github.com/encode/django-rest-framework.git
synced 2024-11-11 12:17:24 +03:00
28 lines
1021 B
Python
28 lines
1021 B
Python
from rest_framework.views import APIView
|
|
from rest_framework import status
|
|
from rest_framework import parsers
|
|
from rest_framework import renderers
|
|
from rest_framework.response import Response
|
|
from rest_framework.authtoken.models import Token
|
|
from rest_framework.authtoken.serializers import AuthTokenSerializer
|
|
|
|
|
|
class ObtainAuthToken(APIView):
|
|
throttle_classes = ()
|
|
permission_classes = ()
|
|
parser_classes = (parsers.FormParser, parsers.MultiPartParser, parsers.JSONParser,)
|
|
renderer_classes = (renderers.JSONRenderer,)
|
|
serializer_class = AuthTokenSerializer
|
|
model = Token
|
|
|
|
def post(self, request):
|
|
serializer = self.serializer_class(data=request.DATA)
|
|
if serializer.is_valid():
|
|
user = serializer.validated_data['user']
|
|
token, created = Token.objects.get_or_create(user=user)
|
|
return Response({'token': token.key})
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
|
|
obtain_auth_token = ObtainAuthToken.as_view()
|