如何在序列化程序创建方法中创建多个对象？

Question

我正在尝试上传一个 csv 文件，然后使用它来填充数据库中的一个表（创建多个对象）。

serializers.py:

def instantiate_batch_objects(data_list, user):
    return [
        WorkData(
            work=db_obj['work'],
            recordTime=db_obj['recordTime'],
            user=user
        ) for db_obj in data_list
    ]


class FileUploadSerializer(serializers.ModelSerializer):
    filedata = serializers.FileField(write_only=True)

    class Meta:
        model = WorkData
        fields = ['user', 'filedata']

    def create(self, validated_data):
        file = validated_data.pop('filedata')
        data_list = csv_file_parser(file)        
        batch = instantiate_batch_objects(data_list, validated_data['user'])
        work_data_objects = WorkData.objects.bulk_create(batch)
        # print(work_data_objects[0])
        return work_data_objects

views.py：

class FileUploadView(generics.CreateAPIView):
    queryset = WorkData.objects.all()
    permission_classes = [IsAuthenticated]
    serializer_class = FileUploadSerializer

    # I guess, this is not need for my case.
    def get_serializer(self, *args, **kwargs):
        print(kwargs.get('data'))
        if isinstance(kwargs.get('data', {}), list):
            kwargs['many'] = True

        return super().get_serializer(*args, **kwargs)

models.py

class WorkData(models.Model):
    user = models.ForeignKey(
        settings.AUTH_USER_MODEL,
        on_delete=models.CASCADE,
        related_name='work_data',
    )
    work = models.IntegerField(blank=False, null=False)
    recordTime = models.DateTimeField(blank=False, null=True)

当我上传文件并发布它时，我收到了这个错误：

尝试在序列化程序 FileUploadSerializer 上获取字段 user 的值时出现 AttributeError。序列化程序字段可能命名不正确，并且与 list 实例上的任何属性或键都不匹配。原始异常文本是：“列表”对象没有属性“用户”。

但我可以看到表已成功填充到数据库中。我应该从FileUploadSerializer 的create 方法返回什么？

Answer 1

好的，在我自己尝试了一个示例之后，我能够重现错误，我对现在发生这种情况的原因有了更好的理解。

首先，我们把create()的实现放到视图类上

def create(self, request, *args, **kwargs):
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        self.perform_create(serializer)
        headers = self.get_success_headers(serializer.data)
        return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)

Got AttributeError when attempting to get a value for field user... 等的原始错误发生是因为 FileUploadView 中的 create() 正在返回 serializer.data，它期望字段 user 和 filedata 但 create() 上的 FileUploadSerializer 是返回一个对象列表，以便您现在了解为什么会发生这种情况。

您可以通过在 FileUploadView 上覆盖 create() 并使用您将创建的 WorkDataSerializer 序列化返回的序列化程序数据来解决此问题

例如：

def create(self, request, *args, **kwargs):
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        self.perform_create(serializer)
        headers = self.get_success_headers(serializer.data)
        workData = WorkDataSerializer(data=serializer.data)
        return Response(workData.data, status=status.HTTP_201_CREATED, headers=headers)

或者，您可以在序列化程序级别执行此操作 - 我更喜欢 -

例如：

class FileUploadSerializer(serializers.ModelSerializer):
    filedata = serializers.FileField(write_only=True)
    created_objects_from_file = serializers.SerializerMethodField()
        
    def get_created_objects_from_file(self, obj):
       file = self.validated_data.pop('filedata')
       data_list = csv_file_parser(file)        
       batch = instantiate_batch_objects(data_list, self.validated_data['user'])
       work_data_objects = WorkData.objects.bulk_create(batch)
       
       return WorkDataSerializer(work_data_objects, many = True).data
    
    
    class Meta:
        model = WorkData
        fields = ['user', 'filedata']



class WorkDataSerializer(serializers.Serializer):
         # fields of WorkData model you want to return

这应该没有问题，注意SerializerMethodField 默认是只读的

见https://www.django-rest-framework.org/api-guide/fields/#serializermethodfield