python - 如何让模型数据在另一个模型的响应中显示为字段
问题描述
这些是我的模型的简化版本(用户模型只是一个 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 的序列化程序中?
解决方案
您设置模型的方式,您可以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')