在 Django Rest Framework 中汇总 ViewSet 结果的总数

Question

我有一个输出一组结果的 ViewSet：

{
    "count": 19355,
    "next": null,
    "previous": null,
    "results": [
        {
            "duration": 5,
            ...
        },
        {
            "duration": 5,
            ...
        },
        ...
    ]
}

我想汇总结果持续时间的总数，就在“计数”旁边那个方便的小顶部区域。我知道如何在查询集中使用 annotate 和 sum 来执行此操作，但我没有看到将其放入 ViewSet 输出的方法。

这个数据集的期望输出是：

{
    "count": 19355,
    "total_duration": 10,
    "next": null,
    "previous": null,
    "results": [
        {
            "duration": 5,
            ...
        },
        {
            "duration": 5,
            ...
        },
        ...
    ]
}

感谢您的帮助！

Answer 1

我建议使用专用的Pagination 类来实现这一点。

class PaginationWithAggregates(pagination.LimitOffsetPagination):
    def paginate_queryset(self, queryset, request, view=None):
        self.total_duration = queryset.aggregate(total_duration=Sum('duration'))['total_duration']
        return super(PaginationWithAggregates, self).paginate_queryset(queryset, request, view)

    def get_paginated_response(self, data):
        paginated_response = super(PaginationWithAggregates, self).get_paginated_response(data)
        paginated_response.data['total_duration'] = self.total_duration
        return paginated_response

别忘了在你的 GenericView 上声明这个分页类。