如何在Django REST Framework中用单个ViewSet创建两个表对象

你可以通过重写PostViewSet的create方法来实现需求，核心逻辑是：成功创建Post后，判断其type是否为"R"，若是则自动创建关联的Review对象。以下是具体实现步骤和代码示例：

1. 重写PostViewSet的create方法 #

假设你的Post模型包含type字段，Review模型通过外键关联Post（如post = models.ForeignKey(Post, on_delete=models.CASCADE)），直接在PostViewSet中重写create方法：

from django.db import transaction
from rest_framework import status
from rest_framework.response import Response
from .models import Post, Review
from .serializers import PostSerializer

class PostViewSet(viewsets.ModelViewSet):
    queryset = Post.objects.all()
    serializer_class = PostSerializer

    @transaction.atomic  # 事务保证Post和Review要么同时创建成功，要么都回滚
    def create(self, request, *args, **kwargs):
        # 验证并保存Post实例
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        post_instance = serializer.save()
        
        # 判断Post类型是否为"R"，是则创建关联Review
        if post_instance.type == "R":
            # 根据你的Review模型字段填充，示例用默认值或从请求数据提取
            Review.objects.create(
                post=post_instance,
                # 可选：如果需要从请求中获取Review内容，可从request.data提取
                # content=request.data.get("review_content", "默认评论"),
                # rating=request.data.get("rating", 3)
            )
        
        # 返回创建成功的响应
        headers = self.get_success_headers(serializer.data)
        return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)

2. 关键细节说明 #

• 事务保障：@transaction.atomic避免出现Post保存成功但Review创建失败的不一致情况，确保操作原子性。
• 直接获取实例：通过serializer.save()直接拿到保存后的Post对象，无需二次查询数据库，提升效率。
• 动态填充Review字段：如果需要用户在创建Post时同时提交Review数据，可从request.data中提取对应字段值，填充到创建Review的参数中。
• 序列化器扩展（可选）：若要在响应中同时返回关联的Review数据，可在PostSerializer中嵌套ReviewSerializer：

  class PostSerializer(serializers.ModelSerializer):
      review = ReviewSerializer(read_only=True)
      
      class Meta:
          model = Post
          fields = ["id", "type", "content", "review"]
  

内容的提问来源于stack exchange，提问作者안금장님