如何使用 DRf 中的嵌套序列化程序和视图集同时更新和创建？

Question

my models:

class User(AbstractUser):
    password = models.CharField(max_length=128, blank=True, null=True)
    email = models.EmailField(max_length=254, unique=True)
    dial_code_id = models.CharField(max_length=100)
    mobile_number = models.CharField(max_length=100, unique=True)
    username = models.CharField(max_length=150, unique=True, blank=True, null=True)
    is_active = models.BooleanField(default=True)

    class Meta:
        db_table = "my_user"


class UserLocation(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE, related_name='location')
    latitude = models.DecimalField(max_digits=8, decimal_places=3)
    longitude = models.DecimalField(max_digits=8, decimal_places=3)
    address = models.TextField()

    class Meta:
        db_table = "user_location"

I have to update 'first_name', 'last_name' and at the same time I have to create an object in the 'UserLocation' table for the same user. For that I have created a viewset

class BasicInformationViewSet(viewsets.ModelViewSet):
  queryset = User.objects.all()
  serializer_class = BasicInformationSerializer

  def list(self, request, *args, **kwargs):
      custom_data = {
          "status": False,
          "message": 'Method Not Allowed',
      }
      return Response(custom_data, status=status.HTTP_200_OK)

  def create(self, request, *args, **kwargs):
      custom_data = {
          "status": False,
          "message": 'Method Not Allowed',
      }
      return Response(custom_data, status=status.HTTP_200_OK)

  def update(self, request, *args, **kwargs):
      instance = self.get_object()
      serializer = self.get_serializer(instance, data=request.data)
      if serializer.is_valid():
          serializer.save()
          custom_data = {
              "status": True,
              "message": 'Successfully added your basic information',
          }
          return Response(custom_data, status=status.HTTP_200_OK)
      else:
          custom_data = {
              "status": False,
              "message": serializer.errors,
          }
          return Response(custom_data, status=status.HTTP_200_OK)

Serializer for the above views:

class UserLocationSerializer(serializers.ModelSerializer):
  latitude = serializers.DecimalField(max_digits=8, decimal_places=3, required=True)
  longitude = serializers.DecimalField(max_digits=8, decimal_places=3, required=True)
  address = serializers.CharField(required=True)

  class Meta:
      model = UserLocation
      fields = "__all__"


class BasicInformationSerializer(serializers.ModelSerializer):
  first_name = serializers.CharField(required=True)
  last_name = serializers.CharField(required=True)
  location = UserLocationSerializer()

  class Meta:
      model = User
      fields = ['first_name', 'last_name', 'location']

  def update(self, instance, validated_data):
      instance.first_name = validated_data.get('first_name', instance.first_name)
      instance.last_name = validated_data.get('last_name', instance.last_name)
      instance.save()
      return instance

The 'first_name' and 'last_name' is getting updated but the fields in 'UserLocation' model is not being updated/created.

Thank you.

Answer 1

modelViewSet jas an attribute http_method_name that say what http method allow to access this class so your code be like:

class BasicInformationViewSet(viewsets.ModelViewSet):
  queryset = User.objects.all()
  serializer_class = BasicInformationSerializer
  http_method_names = ['put']


  def update(self, request, *args, **kwargs):
      instance = self.get_object()
      serializer = self.get_serializer(instance, data=request.data)
      if serializer.is_valid():
          serializer.save()
          custom_data = {
              "status": True,
              "message": 'Successfully added your basic information',
          }
          return Response(custom_data, status=status.HTTP_200_OK)
      else:
          custom_data = {
              "status": False,
              "message": serializer.errors,
          }
          return Response(custom_data, status=status.HTTP_200_OK)

in this code you write i didn't see any block of code that create a UserLocation. however if you have a query that get UserLocation an belong it get User; update user but when access user by UserLocation query you see the old user first name or etc you must use ex: ( user_location.refresh_from_db() ) to update your query of UserLocation.