首页 > 解决方案 > 如何让模型数据在另一个模型的响应中显示为字段

问题描述

这些是我的模型的简化版本(用户模型只是一个 ID 和名称)

class Convo(models.Model):
    owner = models.ForeignKey(User, on_delete=models.CASCADE, related_name='convo_owner')
    users = models.ManyToManyField(User, through='Convo_user')

class Convo_user (models.Model):
    user = models.ForeignKey(UserProfile, on_delete=models.CASCADE)
    convo = models.ForeignKey(Convo, on_delete=models.CASCADE)

class Comments(models.Model):
    name = models.CharField(max_length=255)
    content = models.TextField(max_length=1024)
    convo = models.ForeignKey(Convo, on_delete=models.CASCADE)

这是我的看法

class ConvoViewSet(viewsets.ModelViewSet):

serializer_class = serializers.ConvoSerializer

def get_queryset(self):
    return None

def list(self, request):
    curr_user = request.user.id

    # Collecting the list of conversations
    conversations = models.Conversation.object.filter(ConvoUser__user_id=request.user.id)
    #Getting list of conversation id's
    conv_ids = list(conversations.values_list('id', flat=True).order_by('id'))
    #Getting list of relevant comments
    comments = models.Comments.objects.filter(conversation_id__in=conv_ids)

    return Response(self.get_serializer(conversations, many=True).data)

还有我目前的序列化器

class ConvoSerializer(serializers.ModelSerializer):
"""A serializer for messaging objects"""
# access = AccessSerializer(many=True)

    # model = models.Comments
    # fields = ('id', 'name', 'content', 'convo_id')

class Meta:
    model = models.Convo
    fields = ('id', 'owner_id')

我得到的当前响应是形式

[
    {
        "id": 1,
        "owner_id": 32
    }, ...
]

但是我想在响应中添加一个注释字段,该字段显示注释的所有属性,所以基本上第二个查询集中的所有内容(称为注释),我根本不知道该怎么做。(我以我的方式检索评论,因为我试图最小化对数据库的调用)。我是否需要为评论创建一个新视图,制作自己的序列化程序,然后以某种方式将它们组合到 convo 的序列化程序中?

标签: pythondjango-rest-frameworkdjango-views

解决方案


您设置模型的方式,您可以Convo通过使用 Django 的 ORM访问每个模型的评论convo_object.comments_set.all(),因此您可以设置您ConvoSerializer以访问该实例的评论,如下所示:

class ConvoSerializer(serializers.ModelSerializer):
"""A serializer for messaging objects"""

    comments_set = CommentSerializer(many=True)

    class Meta:
       model = models.Convo
       fields = ('id', 'owner_id', 'comments_set')

然后你定义你的CommentSerializer喜欢:

class CommentSerializer(serializers.ModelSerializer):

    class Meta:
       model = models.Comments
       fields = ('id', 'name', 'content')

推荐阅读