вернуть токен после регистрации в django-rest-framework-simplejwt - PullRequest
Новая
       6

вернуть токен после регистрации в django-rest-framework-simplejwt

0 голосов
/ 27 августа 2018

Я использую django-rest-framework-simplejwt, и мне было интересно, можно ли вернуть токен после регистрации пользователя?

В этом посте есть решение для другого пакета jwtи мне было интересно, как я мог сделать что-то подобное для simplejwt?

спасибо

Ответы [ 3 ]

0 голосов
/ 27 августа 2018

Я только что решил свой вопрос.Дайте мне знать, если у вас есть какие-либо комментарии.Спасибо!

serializers.py 

class RegisterUserSerializer(serializers.ModelSerializer):
    """Serializer for creating user objects."""

    tokens = serializers.SerializerMethodField()

    class Meta:
        model = models.User
        fields = ('id', 'password', 'email', 'tokens')
        extra_kwargs = {'password': {'write_only': True}}

    def get_tokens(self, user):
        tokens = RefreshToken.for_user(user)
        refresh = text_type(tokens)
        access = text_type(tokens.access_token)
        data = {
            "refresh": refresh,
            "access": access
        }
        return data

    def create(self, validated_data):
        user = models.User(
            email=validated_data['email']
        )
        user.set_password(validated_data['password'])
        user.save()    
        return user

views.py 

class UserListView(generics.ListCreateAPIView):
    """Handles creating and listing Users."""
    queryset = User.objects.all()

def create(self, request, *args, **kwargs):
        serializer = RegisterUserSerializer(data=request.data)
        if serializer.is_valid():
            self.perform_create(serializer)
            return Response(serializer.data, status=status.HTTP_201_CREATED)
        return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
0 голосов
/ 06 ноября 2018

другое возможное решение:

по вашему мнению 

from rest_framework_simplejwt.serializers import TokenObtainPairSerializer
from rest_framework_simplejwt.tokens import AccessToken, RefreshToken


@login_required
def index(request):
    tokenr = TokenObtainPairSerializer().get_token(request.user)  
    tokena = AccessToken().for_user(request.user)
    return render(request,'myview/index.html', {"refresh" : str(tokenr),"access" : str(tokena)} )

Я использовал @login_required только для того, чтобы убедиться, что мы аутентифицировали request.user, но вместо этого вы могли бы передать dict

0 голосов
/ 27 августа 2018

Я думаю, вы можете сделать что-то вроде этого: 

def custom_registration_view(request):
    //code to validate & register your user
    payload = jwt_payload_handler(user)
    return HttpResponse(jwt_encode_handler(payload), 201)

Payload_handler, encode_handler и decode_handler, которые вы можете указать в файле настроек.

...