From dee4cf5068bd89bb074d86017f23230d227394bf Mon Sep 17 00:00:00 2001 From: Bruno Villanova Date: Fri, 1 Sep 2017 15:25:53 -0300 Subject: [PATCH] allow use of a custom token model --- rest_framework_expiring_authtoken/views.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/rest_framework_expiring_authtoken/views.py b/rest_framework_expiring_authtoken/views.py index dc7fc97..024673d 100644 --- a/rest_framework_expiring_authtoken/views.py +++ b/rest_framework_expiring_authtoken/views.py @@ -22,14 +22,14 @@ def post(self, request): serializer = AuthTokenSerializer(data=request.data) if serializer.is_valid(): - token, _ = ExpiringToken.objects.get_or_create( + token, _ = self.model.objects.get_or_create( user=serializer.validated_data['user'] ) if token.expired(): # If the token is expired, generate a new one. token.delete() - token = ExpiringToken.objects.create( + token = self.model.objects.create( user=serializer.validated_data['user'] )