解决Django多对多字段直接赋值报错：请使用set()替代

问题原因 #

你在视图代码中直接对Screenshot模型的多对多字段category进行赋值操作（比如category = category或editing_screenshot.category = category），这违反了Django的多对多关系处理规则——多对多关联的集合不能直接赋值，必须通过set()方法来添加或更新关联关系。同时你需要兼容请求中category参数为单个ID或ID列表的情况。

解决方案 #

修改视图中的create和update方法，调整多对多关系的处理逻辑：

1. 修正create方法 #

创建Screenshot实例时不包含category参数，先完成实例创建，再通过set()关联分类：

def create(self, request):
    """ Handle a POST request for a Screenshot item """
    
    archer  = Archer.objects.get(user=request.auth.user)
    captureTool = CaptureTool.objects.get(pk=request.data["captureTool"])
    editingTool = EditingTool.objects.get(pk=request.data["editingTool"])
    
    # 统一处理category参数为列表格式
    category_ids = request.data["category"]
    if not isinstance(category_ids, list):
        category_ids = [category_ids]
    
    # 先创建截图实例，不包含多对多字段
    new_screenshot = Screenshot.objects.create(
        archer=archer,
        image=request.data["image"],
        content=request.data["content"],
        captureTool=captureTool,
        editingTool=editingTool,
        timestamp=request.data["timestamp"]
    )
    
    # 使用set()方法关联分类
    new_screenshot.category.set(category_ids)
    
    serializer = ScreenshotSerializer(new_screenshot)
    return Response(serializer.data, status=status.HTTP_201_CREATED)

2. 修正update方法 #

同样避免直接赋值category，用set()更新关联关系：

def update(self, request, pk):
    """ Handles a PUT request for a Screenshot item """
    
    captureTool = CaptureTool.objects.get(pk=request.data["captureTool"])
    editingTool = EditingTool.objects.get(pk=request.data["editingTool"])
    
    # 统一处理category参数为列表格式
    category_ids = request.data["category"]
    if not isinstance(category_ids, list):
        category_ids = [category_ids]
    
    editing_screenshot = Screenshot.objects.get(pk=pk)
    editing_screenshot.image = request.data["image"]
    editing_screenshot.content = request.data["content"]
    editing_screenshot.captureTool = captureTool
    editing_screenshot.editingTool = editingTool
    editing_screenshot.save()
    
    # 使用set()方法更新分类关联
    editing_screenshot.category.set(category_ids)
    
    return Response(None, status=status.HTTP_204_NO_CONTENT)

3. 配置序列化器返回ID列表 #

确保ScreenshotSerializer正确处理多对多字段，返回你期望的ID列表格式：

from rest_framework import serializers
from .models import Screenshot
from .category import Category

class ScreenshotSerializer(serializers.ModelSerializer):
    category = serializers.PrimaryKeyRelatedField(many=True, queryset=Category.objects.all())
    
    class Meta:
        model = Screenshot
        fields = ['id', 'archer', 'image', 'content', 'captureTool', 'editingTool', 'category', 'timestamp']

测试验证 #

现在可以用Postman发送两种格式的请求体：

• 单个分类：

{
    "archer": 1,
    "image": "https://url.png",
    "content": "content",
    "captureTool": 1,
    "editingTool": 1,
    "category": 1,
    "timestamp": "2020-03-12"
}

• 多个分类：

{
    "archer": 1,
    "image": "https://url.png",
    "content": "content",
    "captureTool": 1,
    "editingTool": 1,
    "category": [1,2,3],
    "timestamp": "2020-03-12"
}

两种请求都会正确创建/更新截图的分类关联，返回结果将符合你期望的JSON格式。

内容的提问来源于stack exchange，提问作者Marcus Antonio Cox