在Django和DRF中控制模型創建

Question

考慮以下Django模型和基於DRF的API：

class Disk(models.Model):
    size = models.IntegerField(...)
    free_space = models.IntegerField(...)


# ---- Serializers ---- #
class DiskSerializer(serializers.ModelSerializer):
    class Meta:
        model = Disk
        fields = ('free_space', 'size')
        read_only_fields = ('size', )


class CreateDiskSerializer(models.Model):
    class Meta:
        model = Disk
        fields = ('size', )  # note, size only!


# ---- View ---- #   
class DisksView(viewsets.ModelViewSet):
    ...

    def get_serializer(self):
        if self.request.method == 'POST':
            return CreateDiskSerializer
        else:
            return DiskSerializer        

我正在嘗試實現在模型創建時根據size初始化free_space字段的值的想法。

天真的方法將覆蓋Disk.save()但這是我最后使用的方法。 另一種方法是創建一個覆蓋.create()的自定義模型管理器，例如：

class DiskManager(models.Manager):
    def create(self, **kwargs):
        if 'free_space' not in kwargs:
            try:
                kwargs['free_space'] = kwargs['size']
            except KeyError:
                pass
        return super().create(**kwargs)

看起來對您來說可以接受嗎？ 有更好的解決方案嗎？

Answer 1

我建議重寫CreateDiskSerializer的create方法，使其看起來像：

def create(self, validated_data):
    validated_data['free_space'] = validated_data['size']
    return super(CreateDiskSerializer, self).create(validated_data)