From ff7cae21eb8e165e7697ced380cb8764a498e764 Mon Sep 17 00:00:00 2001 From: ur001 Date: Thu, 27 Nov 2014 21:36:12 +0300 Subject: [PATCH] Makes it easy to change AuthToken model class when overriding built-in ObtainAuthToken view --- rest_framework/authtoken/views.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rest_framework/authtoken/views.py b/rest_framework/authtoken/views.py index 103abb27c..33f969637 100644 --- a/rest_framework/authtoken/views.py +++ b/rest_framework/authtoken/views.py @@ -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)