Django创建帖子POST接口如何返回含图片视频URL、UUID的完整关联数据

解决方案 #

你可以将单个模型实例包裹为仅包含它自身的QuerySet，直接复用现有工具函数即可，仅需要修改视图中创建资源成功后的返回逻辑：

1. 用Post.objects.filter(pk=post_obj.pk)构造仅包含新创建帖子的查询集
2. 传入你已有的full_post_data_serializer工具函数
3. 取序列化结果列表的第一个元素作为响应数据

修改后的视图对应代码片段如下：

@api_view(['POST'])
def POST_create_post(request):
    serializer = PostSerializer(data=request.data)
    if serializer.is_valid():
        try:
            post_obj = serializer.save()

            try:
                if 'images' in request.FILES.keys():
                    for img in request.FILES.getlist('images'):
                        validate_image_upload(img)
                        Photo.objects.create(post=post_obj, image=img)

                if 'video' in request.FILES.keys():
                    vid = request.FILES.get('video')
                    validate_video_upload(vid)
                    Video.objects.create(post=post_obj, video=vid)
            except ValidationError as e:
                return Response(dict(error=e,
                                     user_message=error_message_generic),
                                status=status.HTTP_400_BAD_REQUEST)

        except django.db.utils.InternalError as e:
            return Response(dict(error=serializer.errors,
                                 user_message=error_message_generic),
                            status=status.HTTP_400_BAD_REQUEST)

        # 替换原有返回逻辑，复用工具函数生成完整序列化数据
        full_post_qs = Post.objects.filter(pk=post_obj.pk)
        full_serializer = full_post_data_serializer(full_post_qs)
        return Response(full_serializer.data[0], status=status.HTTP_201_CREATED)
    else:
        return Response(dict(error=serializer.errors,
                             user_message=error_message_generic),
                        status=status.HTTP_400_BAD_REQUEST)

可选优化 #

如果后续经常需要处理单个对象的序列化，可以对工具函数做兼容改造，避免每次手动取索引：

# helper.py 改造后的工具函数，兼容单个实例和查询集入参
def full_post_data_serializer(post_obj_or_qs):
    if isinstance(post_obj_or_qs, QuerySet):
        query_set_annotated = post_obj_or_qs.annotate(
            creator_username=F('creator__username'),
            goal_description=F('join_goal__goal__description'),
            goal_uuid=F('join_goal__goal__uuid'),
            reply_count=Count('replypost', distinct=True),
            cheer_count=Count('cheerpost', distinct=True)
        ).prefetch_related(
            Prefetch('photo_set', Photo.objects.order_by('-created'))
        )
        return FullPostDataSerializer(query_set_annotated, many=True)
    else:
        # 单个实例先构造查询集做注解处理
        single_qs = Post.objects.filter(pk=post_obj_or_qs.pk)
        query_set_annotated = single_qs.annotate(
            creator_username=F('creator__username'),
            goal_description=F('join_goal__goal__description'),
            goal_uuid=F('join_goal__goal__uuid'),
            reply_count=Count('replypost', distinct=True),
            cheer_count=Count('cheerpost', distinct=True)
        ).prefetch_related(
            Prefetch('photo_set', Photo.objects.order_by('-created'))
        )
        return FullPostDataSerializer(query_set_annotated.first(), many=False)

改造后单个对象可直接传参调用：

full_serializer = full_post_data_serializer(post_obj)
return Response(full_serializer.data, status=status.HTTP_201_CREATED)

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