python – django-rest-framework django-polymorphic ModelSerialization

我想知道是否有人拥有将Django REST框架与django-polymorphic结合在一起的Pythonic解决方案.

鉴于：

class GalleryItem(PolymorphicModel):
    gallery_item_field = models.CharField()

class Photo(GalleryItem):
    custom_photo_field = models.CharField()

class Video(GalleryItem):
    custom_image_field = models.CharField()

如果我想要在django-rest-framework中的所有GalleryItem的列表,它只会给我GalleryItem(父模型)的字段,因此：id,gallery_item_field和polymorphic_ctype.这不是我想要的.如果它是一个视频,我想要custom_photo_field,如果它是一个Photo实例和custom_image_field.

到目前为止,我只测试了GET请求,这样做有效：

class PhotoSerializer(serializers.ModelSerializer):

    class Meta:
        model = models.Photo


class VideoSerializer(serializers.ModelSerializer):

    class Meta:
        model = models.Video


class GalleryItemModuleSerializer(serializers.ModelSerializer):

    class Meta:
        model = models.GalleryItem

    def to_representation(self, obj):
        """
        Because GalleryItem is Polymorphic
        """
        if isinstance(obj, models.Photo):
            return PhotoSerializer(obj, context=self.context).to_representation(obj)
        elif isinstance(obj, models.Video):
           return VideoSerializer(obj, context=self.context).to_representation(obj)
        return super(GalleryItemModuleSerializer, self).to_representation(obj)

对于POST和PUT请求,您可能想要使用to_internal_value def来执行类似于覆盖to_representation定义的类似操作.