Question

我尝试添加自定义创建方法，并使用Django REST框架。创建用户后，我想创建一个身份验证令牌，然后使用此用户和身份验证令牌返回JSON响应。

更新：我能够更新以下内容来创建用户，但现在我得到Cannot assign "<User: User object>": "Token.user" must be a "User" instance我做错了什么？

如何修改以下内容，以便在POST给用户/时，创建用户，创建身份验证令牌并返回两者？

class UserSerializer(serializers.ModelSerializer):
    class Meta:
      model = User
      fields = ('first_name', 'last_name', 'email', 's3_link', 'phone_confirmed', 'agreed_to_tos', 'phone_number', 'facebook_id', 'stripe_id', 'phone_verification_code')

class UserViewSet(viewsets.ModelViewSet):
    queryset = User.objects.all()
    serializer_class = UserSerializer

    ##ISSUE WITH CODE STARTS HERE
    user = serializer.save()
    token = Token.objects.create(user=user)

Answer 1

您可以使用自定义响应将令牌添加到用户数据：

class UserSerializer(serializers.ModelSerializer):
    class Meta:
      model = User
      fields = ('first_name', 'last_name', 'email', 's3_link', 'phone_confirmed', 'agreed_to_tos', 'phone_number', 'facebook_id', 'stripe_id', 'phone_verification_code')

class UserViewSet(viewsets.ModelViewSet):
    queryset = User.objects.all()
    serializer_class = UserSerializer

    # Customized rest_framework.mixins.CreateModelMixin.create():
    def create(self, request, *args, **kwargs):
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)

        # Your code
        user = serializer.save()
        token = Token.objects.create(user=user)

        # Create custom response
        data = serializer.data
        # You may need to serialize your token:
        # token = token.your_token_string_field
        data.update({'token': token})
        headers = self.get_success_headers(serializer.data)
        return Response(data, status=status.HTTP_201_CREATED, headers=headers)

Answer 2

如果您想创建用户，我建议您使用以下格式：

<强> serializers.py：

class AccountCreateSerializer(serializers.ModelSerializer):
    class Meta:
        model = User
        # You will need to check accuracy of fields, but this is for demo purposes
        fields = ['username', 'email', 'password']
        extra_kwargs = {'password': {'write_only': True}}

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

<强> views.py：

from rest_framework import generics
from serializers.py import AccountCreateSerializer

class AccountCreateAPIView(generics.CreateAPIView):
    serializer_class = AccountCreateSerializer

执行Create Always Fails Django REST Framework

2 个答案: