Tivix/django-rest-auth
1
1
Fork 0
mirror of https://github.com/Tivix/django-rest-auth.git synced 2025-04-20 08:42:01 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add JWTSerializerWithExpiration so the expiration dates are serialized correctly

Browse Source
This commit is contained in:
Joel Whitaker 2020-10-29 14:52:30 +00:00 committed by Michael
parent f923bb5a7c
commit 8a227ac60e
2 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
dj_rest_auth/serializers.py
View File

@ -178,6 +178,11 @@ class JWTSerializer(serializers.Serializer):
return user_data
class JWTSerializerWithExpiration(JWTSerializer):
access_token_expiration = serializers.DateTimeField()
refresh_token_expiration = serializers.DateTimeField()
class PasswordResetSerializer(serializers.Serializer):
"""
Serializer for requesting a password reset e-mail.

8
dj_rest_auth/views.py
View File

@ -12,6 +12,7 @@ from rest_framework.permissions import AllowAny, IsAuthenticated
from rest_framework.response import Response
from rest_framework.views import APIView
from serializers import JWTSerializerWithExpiration
from .app_settings import (JWTSerializer, LoginSerializer,
PasswordChangeSerializer,
PasswordResetConfirmSerializer,
@ -51,7 +52,12 @@ class LoginView(GenericAPIView):
def get_response_serializer(self):
if getattr(settings, 'REST_USE_JWT', False):
response_serializer = JWTSerializer
if getattr(settings, 'JWT_AUTH_RETURN_EXPIRATION', False):
response_serializer = JWTSerializerWithExpiration
else:
response_serializer = JWTSerializer
else:
response_serializer = TokenSerializer
return response_serializer