Django DRF и общие отношения: как получить поле content_object из ответа API?
Я реализовал Generic Relations в Django DRF, следуя официальному руководству, и это работает хорошо, за исключением того, что я не могу получить поле content_object из ответа API.
В принципе, у меня есть модель под названием Document, которая может быть связана с моделью под названием Folder или моделью под названием Collection.
class Document(models.Model):
limit = models.Q(app_label='folders', model='folder') | models.Q(
app_label='collections', model='collection')
title = models.CharField(max_length=500)
# Field necessari per la Generic Relation
content_type = models.ForeignKey(
ContentType, on_delete=models.CASCADE, null=True, blank=True, limit_choices_to=limit)
object_id = models.PositiveIntegerField(null=True, blank=True)
content_object = GenericForeignKey(
'content_type', 'object_id')
category = models.CharField(max_length=30, blank=True, null=True)
def __str__(self):
return self.title
class Folder(models.Model):
...
documents = GenericRelation(Document)
def __str__(self):
return self.title
class Collection(models.Model):
...
documents = GenericRelation(Document)
def __str__(self):
return self.title
Вот мои сериализаторы:
class ContentObjectRelatedField(serializers.RelatedField):
def to_representation(self, value):
if isinstance(value, Folder):
serializer = FolderSerializer(value)
elif isinstance(value, Collection):
serializer = CollectionSerializer(value)
else:
raise Exception('Unexpected type of object')
return serializer.data
class DocumentSerializer(serializers.ModelSerializer):
class Meta:
model = Document
fields = ('id', 'title', 'content_type',
'object_id', 'category')
class FolderSerializer(serializers.ModelSerializer):
documents = DocumentSerializer(many=True, read_only=True)
class Meta:
model = Folder
fields = ("id", "title", "description",
"documents")
depth = 1
(Collection serializer is essentially the same ad the Folder serializer, with its own fields).
Я ожидал, что смогу получить доступ к содержимому поля content_object при извлечении - с помощью GET-запроса к конечной точке API - документов. Вместо этого это поле недоступно. Если я пытаюсь добавить его к полям, перечисленным в сериализаторах, возникает ошибка.
Как я могу получить доступ к этому содержимому, чтобы знать для каждого документа, к какой папке или какой коллекции он принадлежит?
Большое спасибо.
Попробуйте это:
class ContentObjectRelatedField(serializers.RelatedField):
def to_representation(self, value):
if isinstance(value, Folder):
serializer = FolderForDocumentSerializer(value)
elif isinstance(value, Collection):
serializer = CollectionForDocumentSerializer(value) # Defines CollectionForDocumentSerializer in the same manner of FolderForDocumentSerializer
else:
raise Exception('Unexpected type of object')
return serializer.data
class FolderForDocumentSerializer(serializers.ModelSerializer):
class Meta:
model = Folder
fields = ("id", "title", "description")
depth = 1
class DocumentSerializer(serializers.ModelSerializer):
content_object = ContentObjectRelatedField(read_only=True)
class Meta:
model = Document
fields = ('id', 'title', 'content_object', 'category')
# Note that you can use DocumentSerializer and CollectionSerializer, but not in ContentObjectRelatedField.to_representation
Ваш фронтенд может определить тип объекта content_object, исследуя возвращаемые поля