Django Rest Framework - 自动注释查询集

Question

我在项目的许多不同地方都使用了一个Serializer。我需要使用一个注释，但问题是我不想在所有视图中都对其进行注释，所以我想在 Serializer 本身中进行通用注释。

有可能吗？

现在我需要在每次序列化之前这样做：

City.objects....filter....annotate(
                number_of_users_here_now=Count('current_userprofiles'))

我试过了：

class NumberOfUsersInCityNowField(serializers.Field):
    def to_native(self, value):
        count = value.annotate(
            number_of_users_here_now=Count('current_userprofiles'))['current_userprofiles__count']
        return count


class CityMapSerializer(serializers.ModelSerializer):
    number_of_users_here_now = NumberOfUsersInCityNowField()

    class Meta:
        model = City
        fields = ('place_id', 'lat', 'lng', 'number_of_users_here_now', 'formatted_address')

这个Serializer 返回：

/api/ajax-check-trip-creation 处的 AttributeError 得到 AttributeError 尝试获取字段 number_of_users_here_now 的值时 序列化程序CityMapSerializer。序列化器字段可能被命名 不正确且不匹配 City 实例上的任何属性或键。 原始异常文本是：“城市”对象没有属性 'number_of_users_here_now'。

编辑

class NumberOfUsersInCityNowField(serializers.PrimaryKeyRelatedField):
    def get_queryset(self):
        return City.objects.annotate(
        number_of_users_here_now=Count('current_userprofiles'))

class CityMapSerializer(serializers.ModelSerializer):
    # number_of_users_here_now = serializers.IntegerField()
    number_of_users_here_now = NumberOfUsersInCityNowField()
    class Meta:
        model = City
        fields = ('place_id', 'lat', 'lng', 'number_of_users_here_now', 'formatted_address')

但是

serializers.CityMapSerializer(City.objects.all()[:3],many=True).data

仍然返回：

AttributeError: 'City' object has no attribute 'number_of_users_here_now'

Answer 1

您可以将查询集的count 方法与SerializerMethodField 一起使用：

class CityMapSerializer(serializers.ModelSerializer):
    number_of_users_here_now = SerializerMethodField()

    def get_number_of_users_here_now (self, obj):
        return obj.current_userprofiles.count()

UPD

为了避免 n+1 查询，您可以尝试实现NumberOfUsersInCityNowField 序列化程序的get_queryset 方法：

class NumberOfUsersInCityNowField(serializers.PrimaryKeyRelatedField):
    def get_queryset(self):
        return City.objects.annotate(
        number_of_users_here_now=Count('current_userprofiles'))['current_userprofiles__count']