Skip to content

Commit 181fa26

Browse files
authored
Merge pull request #170 from pythonkr/feature/include-only-paid
fix: 후원금 미입금 상태인 후원사는 표시되지 않도록 수정
2 parents 2ab18db + 203219d commit 181fa26

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

sponsor/viewsets.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22

33
from django.db.transaction import atomic
44

5+
from django.db.models import Prefetch
56
from django.shortcuts import get_object_or_404
67
from django.db.utils import IntegrityError
78
from rest_framework import mixins, status, viewsets
@@ -53,9 +54,12 @@ def get_serializer_class(self):
5354

5455
@action(detail=False, methods=["GET"], url_path="with-sponsor")
5556
def list_with_levels(self, request, version):
56-
serializer = self.get_serializer(
57-
self.get_queryset().filter(sponsor__paid_at__isnull=False), many=True
57+
queryset = self.get_queryset().prefetch_related(
58+
Prefetch(
59+
"sponsor_set", queryset=Sponsor.objects.filter(paid_at__isnull=False)
60+
)
5861
)
62+
serializer = self.get_serializer(queryset, many=True)
5963
return Response(serializer.data)
6064

6165
@action(detail=False, methods=["POST"])

0 commit comments

Comments
 (0)