DRF中如何创建支持批量外键ID的POST序列化器

问题背景 #

现有两个Django模型：

class Skill(models.Model):
    name = models.CharField(max_length=100)

    def __str__(self):
        return self.name + " - ID: " + str(self.id)

class Experience(models.Model):
    consultant = models.ForeignKey("Consultant", related_name="experience", on_delete=models.CASCADE)
    project_name = models.CharField(max_length=100)
    company = models.CharField(max_length=100)
    company_description = models.TextField(null=True, blank=True)
    from_date = models.DateField()
    to_date = models.DateField()
    project_description = models.CharField(max_length=100)
    contribution = models.TextField()
    summary = models.TextField()
    is_pinned = models.BooleanField(default=False)
    role = models.CharField(max_length=100, null=True)
    skill = models.ForeignKey("Skill", related_name="experience", on_delete=models.CASCADE)

需要实现/experience/的POST接口，支持传入Skill外键ID列表（如skill_ids: [1,2,3]），满足：

• 若所有ID对应的Skill存在，生成3条Experience记录（每条对应一个Skill）
• 若存在无效ID，在验证阶段返回错误提示

之前的错误分析 #

错误1：PrimaryKeyRelatedField写法触发非空约束 #

class ExperienceSerializer(serializers.ModelSerializer):
    skills = serializers.PrimaryKeyRelatedField(
        many=True,
        queryset=Skill.objects.all(),
        write_only=True
    )

    class Meta:
        model = Experience
        exclude = ['skill']

    def create(self, validated_data):
        skills_data = validated_data.pop('skills', [])
        experience = Experience.objects.create(**validated_data)

        for skill in skills_data:
            experience.skill.add(skill)

        return experience

报错：django.db.utils.IntegrityError: null value in column "skill_id" violates not-null constraint

错误原因：

• Experience的skill是必填ForeignKey，创建时必须指定，代码中先创建无skill的Experience，直接触发非空约束
• 混淆了ForeignKey与ManyToMany关系：ForeignKey是一对一关联（每个Experience对应一个Skill），没有add方法，无法给单条Experience关联多个Skill

错误2：ListField+SkillSerializer触发400错误 #

class ExperienceSerializer(serializers.ModelSerializer):
    skill_ids = serializers.ListField(
        child=SkillSerializer(),
        write_only=True
    )

    class Meta:
        model = Experience
        fields = (
            'consultant',
            'project_name',
            'company',
            'company_description',
            'from_date',
            'to_date',
            'project_description',
            'contribution',
            'summary',
            'is_pinned',
            'role',
            'skill',
            'skill_ids'
        )

    def create(self, validated_data):
        skill_ids = validated_data.pop('skill_ids')
        experience = Experience.objects.create(**validated_data)
        experience.set(skill_ids)

        return experience

报错：

{
    "skill": ["This field is required."],
    "skill_ids": {
        "0": {"non_field_errors": ["Invalid data. Expected a dictionary, but got int."]}
    }
}

错误原因：

• SkillSerializer期望接收Skill对象的字典结构（如{"name": "Python"}），但前端传入的是整数ID，类型不匹配
• 序列化器包含了必填字段skill，但前端未传入，触发必填校验

正确解决方案 #

1. 序列化器实现 #

from rest_framework import serializers
from .models import Experience, Skill

class ExperienceSerializer(serializers.ModelSerializer):
    skill_ids = serializers.ListField(
        child=serializers.PrimaryKeyRelatedField(queryset=Skill.objects.all()),
        write_only=True
    )

    class Meta:
        model = Experience
        exclude = ['skill']  # 排除skill字段，避免前端传入要求

    def create(self, validated_data):
        # 取出验证通过的Skill对象列表
        skills = validated_data.pop('skill_ids')
        experiences = []
        
        # 循环每个Skill，创建对应的Experience记录
        for skill in skills:
            experience = Experience.objects.create(skill=skill, **validated_data)
            experiences.append(experience)
        
        return experiences

关键细节解释：

• skill_ids字段：用ListField包裹PrimaryKeyRelatedField，既验证输入是列表格式，又自动校验每个ID对应的Skill是否存在（无效ID会在验证阶段直接返回错误，无需手动处理）
• exclude=['skill']：模型中skill是必填项，但我们通过skill_ids自动赋值，所以排除该字段，避免DRF强制要求前端传入
• create方法：循环创建每条Experience，每条记录都关联对应的Skill，完全符合模型的非空约束

2. 自定义视图处理批量返回 #

默认的CreateAPIView只支持返回单个实例，需要自定义视图处理批量创建后的结果：

from rest_framework import generics, status
from rest_framework.response import Response
from .models import Experience
from .serializers import ExperienceSerializer

class ExperienceCreateView(generics.CreateAPIView):
    serializer_class = ExperienceSerializer

    def create(self, request, *args, **kwargs):
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        experiences = serializer.save()
        
        # 序列化所有创建的记录并返回
        response_serializer = ExperienceSerializer(experiences, many=True)
        return Response(response_serializer.data, status=status.HTTP_201_CREATED)

视图说明：

• 调用序列化器的save()方法得到创建的Experience列表
• 用many=True序列化列表，返回201状态码符合REST规范

3. 测试请求示例 #

请求体：

{
  "project_name": "Project AVC",
  "company": "XYZ Company",
  "company_description": "Description of XYZ Company",
  "from_date": "2022-01-01",
  "to_date": "2022-12-31",
  "project_description": "Description of Project ABC",
  "contribution": "Contributions to Project ABC",
  "summary": "Summary of Experience",
  "is_pinned": false,
  "role": "Consultant",
  "skill_ids": [1,2,3],
  "consultant": 1
}

• 若ID 1、2、3的Skill都存在，会创建3条Experience记录，每条的skill_id分别对应1、2、3
• 若存在无效ID（如4），验证阶段会返回：

  {
      "skill_ids": {
          "3": ["Invalid pk \"4\" - object does not exist."]
      }
  }
  

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