Facilitate replacing AuthToken model

knox/auth.py

@@ -62,7 +62,7 @@ def authenticate_credentials(self, token):
         '''
         msg = _('Invalid token.')
         token = token.decode("utf-8")
-        for auth_token in AuthToken.objects.filter(
+        for auth_token in self.model.objects.filter(
                 token_key=token[:CONSTANTS.TOKEN_KEY_LENGTH]):
             if self._cleanup_token(auth_token):
                 continue

knox/models.py

@@ -22,7 +22,9 @@ def create(self, user, expiry=knox_settings.TOKEN_TTL):
         return instance, token
 
 
-class AuthToken(models.Model):
+class AbstractAuthToken(models.Model):
+    class Meta:
+        abstract = True
 
     objects = AuthTokenManager()
 
@@ -37,3 +39,7 @@ class AuthToken(models.Model):
 
     def __str__(self):
         return '%s : %s' % (self.digest, self.user)
+
+
+class AuthToken(AbstractAuthToken):
+    pass

knox/views.py

@@ -15,6 +15,7 @@
 class LoginView(APIView):
     authentication_classes = api_settings.DEFAULT_AUTHENTICATION_CLASSES
     permission_classes = (IsAuthenticated,)
+    model = AuthToken
 
     def get_context(self):
         return {'request': self.request, 'format': self.format_kwarg, 'view': self}
@@ -60,7 +61,7 @@ def post(self, request, format=None):
                     status=status.HTTP_403_FORBIDDEN
                 )
         token_ttl = self.get_token_ttl()
-        instance, token = AuthToken.objects.create(request.user, token_ttl)
+        instance, token = self.model.objects.create(request.user, token_ttl)
         user_logged_in.send(sender=request.user.__class__,
                             request=request, user=request.user)
         data = self.get_post_response_data(request, token, instance)