QuerySet 메소드 aggregate() 활용에 대하여

강사님 안녕하세요~
QuerySet 메소드 중 aggregate()를 활용하는데 어려움이 있어서 도움을 요청드립니다.

aggregate()를 활용하여 QuerySet 내 모델 객체에 대한 다양한 집계결과를 하나의 JSON 객체로 반환하고 싶습니다.

다음과 같이 응답을 주고 싶습니다.

{

"total_count": 3,

"estrus_count": 2,

"pregnancy_count": 0

}

문제는 aggregate()의 반환값이 모델로 구성된 QuerySet이 아니라 Dictionary이기 때문에 viewset의 aggregate()로 처리된 값을 Serializer에서 받아서 JSON으로 반환할 때 관련 에러가 계속 발생합니다.

viewset에서 aggregate로 작업한 결과를 받아서 serializer에서 의도한대로 하나의 JSON 객체로 반환할 수 있는 방법이 있을까요?

Model, Serializer, ViewSet에 대한 코드는 아래와 같습니다.

class DogStatusCountViewSet(viewsets.ModelViewSet):
    serializer_class = DogStatusCountSerializer
    http_method_names = ['get', ]
    ordering_fields = '__all__'

    def get_queryset(self):
        queryset = super().get_queryset()

        return queryset.aggregate(
            total_count=Count('id'),
            estrus_count=Count('id', filter=Q(current_status='estrus')),
            pregnancy_count=Count('id', filter=Q(current_status='pregnancy'))
        )

class DogStatusCountSerializer(serializers.Serializer):
    total_count = serializers.IntegerField(read_only=True)
    estrus_count = serializers.IntegerField(read_only=True)
    pregnancy_count = serializers.IntegerField(read_only=True)

class Dog(models.Model):
    current_status = models.CharField(max_length=64, default='estrus')

    class Meta:
        managed = True
        db_table = 'dog'
        ordering = ['-id']

 
감사합니다

인프런 커뮤니티 질문&답변