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, исследуя возвращаемые поля

Вернуться на верх