将计数字段添加到 django rest 框架序列化程序

发布于10月26日

我正在序列化内置的Django Group模型，并希望在序列化程序中添加一个字段来计算组中的用户数.我目前正在使用以下序列化程序:

class GroupSerializer(serializers.ModelSerializer):
    class Meta:
        model = Group
        fields = ('id', 'name', 'user_set')

这将返回组ID和名称以及组中的用户数组(用户ID):

{
    "id": 3,
    "name": "Test1",
    "user_set": [
      9
    ]
}

相反，我希望作为输出的内容是这样的:

{
    "id": 3,
    "name": "Test1",
    "user_count": 1
}

任何帮助都将不胜感激.谢谢.

推荐答案

这应该管用

class GroupSerializer(serializers.ModelSerializer):

    user_count = serializers.SerializerMethodField()

    class Meta:
        model = Group
        fields = ('id', 'name','user_count')

    def get_user_count(self, obj):
        return obj.user_set.count()

这将向序列化程序添加一个user_count字段，该字段的值由get_user_count设置，它将返回user_set的长度.

您可以在此处找到有关SerializerMethodField的更多信息:http://www.django-rest-framework.org/api-guide/fields/#serializermethodfield