如何正确扩展Django ModelSerializer的create方法实现自动填充模型字段

现有代码的问题 #

你当前编写的create方法存在两个明显错误：

• 重复创建实例：super().create(validated_data)已经完成了一次实例保存逻辑，后续再调用ImageUpload.objects.create会生成两条重复的无效数据
• 字段类型不匹配：width和height在模型中定义为IntegerField，你赋值的是字符串类型，保存时会触发类型校验错误

正确扩展方式 #

你完全可以通过super()调用原生create方法实现扩展，不需要重写全部逻辑，有两种常用实现方案：

方案1：提前修改validated_data（推荐） #

在调用父类create方法前，把需要自动补充的字段直接写入validated_data，父类原生逻辑会直接用修改后的参数创建实例，全程只执行一次数据库写入，性能最优。
正确代码如下：

from PIL import Image
from rest_framework import serializers
from .models import ImageUpload

class ImageUploadSerializer(serializers.ModelSerializer):
    class Meta:
        model = ImageUpload
        # 仅保留前端需要传入的字段，其余自动填充字段设为只读，避免校验报错
        fields = ['picture']
        read_only_fields = ['name', 'url', 'width', 'height', 'parent_picture']

    def create(self, validated_data):
        # 1. 提取上传的图片文件对象
        img_file = validated_data['picture']
        # 2. 自动计算图片宽高
        with Image.open(img_file) as img:
            width, height = img.size
        # 3. 补充需要自动赋值的字段到validated_data
        validated_data.update({
            'name': img_file.name,
            'width': width,
            'height': height,
            'url': img_file.url # 直接用ImageField自带的url属性
        })
        # 4. 调用父类原生create方法完成保存，直接返回结果即可
        return super().create(validated_data)

方案2：实例创建后补全字段再保存 #

如果你的自动赋值逻辑需要依赖实例创建后的属性（比如自增ID），可以用这种方式：

def create(self, validated_data):
    # 先调用父类方法创建实例
    instance = super().create(validated_data)
    # 补全字段
    img_file = instance.picture
    with Image.open(img_file) as img:
        instance.width, instance.height = img.size
    instance.name = img_file.name
    instance.url = img_file.url
    # 手动保存修改
    instance.save()
    return instance

补充说明 #

如果需要用到PIL读取图片宽高，需要先安装Pillow依赖：

pip install pillow

内容的提问来源于stack exchange，提问作者Vlad Vlad