你能帮我解决Django rest框架中的泛型关系的序列化问题吗?我需要获取 SWOT 对象的数据,包括公司及其所有评论。我在检索属于通用字段的评论时遇到问题。
模型.py:
class SWOT(models.Model):
name = models.CharField(max_length=500, default='')
description = models.CharField(max_length=500, default='', blank=True)
company = models.ForeignKey(Company, on_delete=models.CASCADE)
#Generic Relation for comments
comments = GenericRelation(Comment)
def __str__(self):
return self.name
class Comment(models.Model):
author = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE)
text = models.TextField(verbose_name="")
created_date = models.DateTimeField(default=timezone.now)
content_type = models.ForeignKey(ContentType, on_delete=models.CASCADE)
object_id = models.PositiveIntegerField()
content_object = GenericForeignKey()
视图.py:
class SwotView(generics.ListAPIView):
serializer_class = SwotSerializer
def get_queryset(self):
swot_id = self.kwargs['swot_id']
return SWOT.objects.filter(id=swot_id)
序列化程序.py:
class CommentRelatedField(serializers.RelatedField):
def to_representation(self, value):
if isinstance(value, SWOT):
serializer = SwotSerializer(value)
else:
raise Exception('Unexpected type of comment ', type(value))
return serializer.data
class SwotSerializer(serializers.ModelSerializer):
company = CompanySerializer()
comments = CommentRelatedField(many=True, queryset=Comment.objects.all())
class Meta:
model = SWOT
fields = ('id','name', 'description','company','comments')
我试图实现文档中示例中的代码,但我解决了错误。我不知道为什么对象的类型是Comment而不是 SWOT。
你能告诉我我能做些什么来使这段代码像在文档中一样工作,或者像文档建议的那样提出另一种解决方案,比如常规的关系字段。
请注意,使用 GenericRelation 字段表示的反向通用键可以使用常规关系字段类型进行序列化,因为关系中目标的类型始终是已知的。
先感谢您。