Question

考虑这三个模型：

class Genre(models.Model):
    slug = models.CharField(max_length=30, unique=True)
    name = models.CharField(max_length=300)

class Group(models.Model):
    slug = models.CharField(max_length=30, unique=True)
    name = models.CharField(max_length=300)
    genre = models.ForeignKey('Genre', related_name='groups')

class Album(models.Model):
    name = models.CharField(max_length=300)
    track_count = models.IntegerField()
    artist = models.ForeignKey('Group', related_name='albums')

现在，我需要这种DRF输出：

[ 
  {'name': 'Rock', 'albums': [
         {'name': 'Meteora', 'track_count': 12}, 
         {'name': 'Master of Puppets', 'track_count': 10'}]
  },
  {..},
]

换句话说，我需要获得每个类型的所有专辑，绕过“组”模式。我已经通过这种方式取得了类似的成就：

views.py

class SegmentViewSet(viewsets.ModelViewSet):

    queryset = Genre.objects.all().annotate(Count('name'))
    serializer_class = GenreSerializer

serializers.py

class GroupSerializer(serializers.ModelSerializer):
    albums = AlbumSerializer(many=True, read_only=True)

    def to_representation(self, value):
        return value.products.annotate(Count('id')).values('name', 'sap_code')

class Meta:
    model = Group
    fields = ('albums',)


class GenreSerializer(serializers.ModelSerializer):
    albums = GroupSerializer(many=True, read_only=True)

    class Meta:
        model = Genre
        fields = ('id',
                  'name',
                  'albums')

这给了我这样的东西：

[
  {'name': 'Rock', 'albums': [
      [ 
        {'name': 'Meteora', 'track_count': 12}, 
        {...}, 
        {...} 
      ], 
      [ 
        {...}, 
        {...}
      ] 
    ] 
  }
]

这是专辑，按专辑分组，按流派分组。这是一个嵌套数组，我只需要在一个数组中按类型分组的专辑。

我知道我的to_representation函数的解决方案非常脏，但它与我正在寻找的东西相似。但是，嘿，这就是自定义相关领域的用途。

有没有办法在没有直接将专辑链接到流派的情况下获得所需的输出？确实有理由保持这种方式。

我已经尝试过自定义相关字段，PrimaryKeyRelatedField，甚至试图在我的视图中重载get_queryset。我忽视了什么吗？

Answer 1

您可以尝试按如下方式定义序列化程序：

from rest_framework import serializers


class GenreSerializer(serializers.ModelSerializer):
    albums = serializers.SerializerMethodField()

    class Meta:
        model = Genre

    def get_albums(self, obj):
        albums = Album.objects.filter(group__genre=obj)
        albums_serializer = AlbumSerializer(data=albums, many=True)
        albums_serializer.is_valid()
        return albums_serializer.data


class AlbumSerializer(serializers.ModelSerializer):
    class Meta:
        model = Album

结果应如下：

[{
        'name': 'Rock',
        'albums': [{
                'name': 'Meteora',
                'track_count': 13
            },
            {
                'name': 'Master of Puppets',
                'track_count': 12
            },
            {
                'name': 'Nevermind',
                'track_count': 11
            }
        ]
    },
    {
        'name': 'Pop',
        'albums': [{
                'name': 'Ray of Light',
                'track_count': 11
            },
            {
                'name': 'Thriller',
                'track_count': 14
            }
        ]
    }
]

请注意，这是高度未经优化和未经测试的，但从此处我确定您知道如何在视图中使用GenreSeralizer并优化代码。

Django在嵌套关系中休息框架绕过实体

1 个答案: