rest_framework_simplejwt custom tokengenerator
from rest_framework import serializers, permissions, status
from rest_framework.generics import CreateAPIView
from rest_framework.response import Response
from rest_framework.validators import UniqueValidator
from rest_framework_simplejwt.serializers import TokenObtainPairSerializer
from bp_auth.models import CustomUser as User
class SignupSerializer(serializers.ModelSerializer):
password = serializers.CharField(write_only=True)
email = serializers.CharField(
validators=[UniqueValidator(queryset=User.objects.all())]
)
class Meta:
model = User
fields = ('email', 'password')
class MyTokenObtainPairSerializer(TokenObtainPairSerializer):
@classmethod
def get_token(cls, user):
token = super(MyTokenObtainPairSerializer, cls).get_token(user)
token['email'] = user.email
return token
class RegisterUsers(CreateAPIView):
serializer_class = SignupSerializer
permission_classes = (permissions.AllowAny,)
def post(self, request, *args, **kwargs):
serialized_user = SignupSerializer(data=request.data)
if serialized_user.is_valid():
User.objects.create_user(
serialized_user.initial_data['email'],
serialized_user.initial_data['password']
)
tokens = MyTokenObtainPairSerializer(request.data).validate(request.data)
return Response(tokens, status=status.HTTP_201_CREATED)
else:
return Response(serialized_user._errors, status=status.HTTP_400_BAD_REQUEST)