Django Rest Framework多对多关系序列化问题：创建电影并关联流派

嘿，我懂你在DRF里处理多对多关系时的头疼！你现在要创建电影并关联流派，对吧？先别急，咱们一步步来解决这个问题。首先我注意到你给出的Movie模型里还没明确写出和Genre的多对多关联字段，先把这个关键部分补全：

class Genre(models.Model):
    id = models.IntegerField(primary_key=True)
    name = models.CharField(null=False,blank=False,max_length=50)
    
    def __str__(self):
        return self.name

class Movie(models.Model):
    id = models.IntegerField(primary_key=True,null=False,blank=False)
    adult = models.BooleanField(null=False,blank=False)
    backdrop_path = models.CharField(null=True,blank=True,max_length=200)
    budget = models.BigIntegerField (default=0)
    homepage = models.CharField(null=True,blank=True,max_length=200)
    original_language = models.CharField(max_length=10)
    original_title = models.CharField(max_length=200)
    overview = models.CharField(max_length=1000)
    poster_path = models.CharField(null=True,blank=True,max_length=200)
    # 补充多对多关联字段，related_name方便从Genre反向查询关联的电影
    genres = models.ManyToManyField(Genre, related_name='movies')
    
    def __str__(self):
        return self.original_title

接下来是序列化器的问题——DRF默认的ModelSerializer在创建对象时不会自动处理多对多字段，所以咱们需要重写create方法来手动处理关联逻辑。这里分两种场景给你方案：

场景1：创建电影时同时创建/关联流派（支持传入完整流派信息） #

如果你的需求是创建电影时，既可以关联已有的流派，也能创建新的流派，那可以用嵌套序列化器：

from rest_framework import serializers
from .models import Movie, Genre

class GenreSerializer(serializers.ModelSerializer):
    class Meta:
        model = Genre
        fields = ['id', 'name']

class MovieSerializer(serializers.ModelSerializer):
    # 用嵌套序列化器接收流派的完整信息
    genres = GenreSerializer(many=True)

    class Meta:
        model = Movie
        fields = ['id', 'adult', 'backdrop_path', 'budget', 'homepage', 'original_language', 'original_title', 'overview', 'poster_path', 'genres']

    def create(self, validated_data):
        # 先把流派数据从验证后的字典里取出，因为Movie的create方法不处理多对多字段
        genres_data = validated_data.pop('genres')
        # 创建电影对象
        movie = Movie.objects.create(**validated_data)
        # 遍历流派数据，用get_or_create避免重复创建已存在的流派
        for genre_data in genres_data:
            genre, _ = Genre.objects.get_or_create(id=genre_data['id'], defaults={'name': genre_data['name']})
            movie.genres.add(genre)
        return movie

场景2：仅关联已存在的流派（只需传入流派ID） #

如果你的流派已经提前创建好了，只需要在创建电影时关联它们，那可以用PrimaryKeyRelatedField，这种情况不需要重写create方法，DRF会自动处理：

from rest_framework import serializers
from .models import Movie, Genre

class MovieSerializer(serializers.ModelSerializer):
    # 直接接收流派ID列表，queryset用于验证ID是否存在
    genres = serializers.PrimaryKeyRelatedField(many=True, queryset=Genre.objects.all())

    class Meta:
        model = Movie
        fields = ['id', 'adult', 'backdrop_path', 'budget', 'homepage', 'original_language', 'original_title', 'overview', 'poster_path', 'genres']

最后是视图层，用最简单的CreateAPIView或者自定义APIView都可以，比如：

from rest_framework.views import APIView
from rest_framework.response import Response
from rest_framework import status
from .serializers import MovieSerializer

class MovieCreateView(APIView):
    def post(self, request):
        serializer = MovieSerializer(data=request.data)
        if serializer.is_valid():
            serializer.save()
            return Response(serializer.data, status=status.HTTP_201_CREATED)
        return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)

举个请求例子 #

如果是场景1，POST请求的JSON数据大概是这样：

{
    "id": 1,
    "adult": false,
    "backdrop_path": "/test/backdrop.jpg",
    "budget": 150000000,
    "homepage": "https://example.com/avengers",
    "original_language": "en",
    "original_title": "The Avengers",
    "overview": "Super heroes team up to save the world.",
    "poster_path": "/test/poster.jpg",
    "genres": [
        {"id": 28, "name": "Action"},
        {"id": 12, "name": "Adventure"}
    ]
}

如果是场景2，请求数据里的genres直接传ID列表就行：

{
    // 其他字段同上
    "genres": [28, 12]
}

备注：内容来源于stack exchange，提问作者12341413451354134