Makes it easy to change AuthToken model class when overriding built-in ObtainAuthToken view

This commit is contained in:
ur001 2014-11-27 21:36:12 +03:00
parent d2d7e1dfde
commit ff7cae21eb

View File

@ -19,7 +19,7 @@ class ObtainAuthToken(APIView):
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)
token, created = self.model.objects.get_or_create(user=user)
return Response({'token': token.key})
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)