Skip to content

Commit dee4cf5

Browse files
allow use of a custom token model
1 parent e62f1f9 commit dee4cf5

File tree

1 file changed

+2
-2
lines changed
  • rest_framework_expiring_authtoken

1 file changed

+2
-2
lines changed

rest_framework_expiring_authtoken/views.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,14 +22,14 @@ def post(self, request):
2222
serializer = AuthTokenSerializer(data=request.data)
2323

2424
if serializer.is_valid():
25-
token, _ = ExpiringToken.objects.get_or_create(
25+
token, _ = self.model.objects.get_or_create(
2626
user=serializer.validated_data['user']
2727
)
2828

2929
if token.expired():
3030
# If the token is expired, generate a new one.
3131
token.delete()
32-
token = ExpiringToken.objects.create(
32+
token = self.model.objects.create(
3333
user=serializer.validated_data['user']
3434
)
3535

0 commit comments

Comments
 (0)