如何return特定字段queryset DRF

Question

我想创建一个自定义查询集 class return 的不同字段来预定义两种情况。

当 DateField 大于今天时
不到今天的时候。

如果它大于 return 所有字段，否则 return 仅 date_to_open 和 post_name 字段。

views.py

class GroupDetail(generics.RetrieveAPIView):
    serializer_class = serializers.GroupDetailsSerializer
    permission_classes = (IsAuthenticated, )

    def greater(self):
        return models.Group.objects.filter(shared_to=self.request.user, 
           date_to_open__gt=timezone.now()).exists()

 

    def get_queryset(self, *args, **kwargs):
        if self.greater():
            query_set = models.Group.objects.filter(shared_to=self.request.user, 
               date_to_open__gt=timezone.now())
        else:
            query_set = SPECIFIC FIELDS
        return query_set

serializers.py

class GroupDetailsSerializer(serializers.ModelSerializer):

    owner = serializers.ReadOnlyField(source='owner.name')
    images = GroupImageSerializer(many=True, read_only=True)
    shared_to = serializers.SlugRelatedField(queryset=models.UserProfile.objects.all(), 
       slug_field='name', many=True)

    class Meta:
        model = models.Group
        fields = ('id', 'group_name', 'owner', 'group_text', 'created_on', 'date_to_open', 'shared_to', 
    'images', )

Answer 1

您可以继续使用相同的逻辑并使用 values_list 到 return 查询集中的特定值。 returned 值也是一个查询集

def get_queryset(self, *args, **kwargs):
    if self.greater():
        return models.Group.objects.filter(shared_to=self.request.user, date_to_open__gt=timezone.now())
    else:
        return models.Group.objects.filter(shared_to=self.request.user, date_to_open__lt=timezone.now()).values_list('date_to_open', 'post_name' , flat = True)

Answer 2

好的。感谢@ArakkalAbu 的评论，我刚刚覆盖了 get_serializer_class()

views.py

class GroupDetail(generics.RetrieveAPIView):
    queryset = models.Group.objects.all()
    serializer_class = serializers.GroupDetailsSerializer
    permission_classes = (IsAuthenticated, )

    def greater(self):
        return models.Group.objects.filter(shared_to=self.request.user, date_to_open__gt=timezone.now()).exists()


    def get_serializer_class(self):
        if self.greater():
            return serializers.GroupDetailsSerializer
        else:
            return serializers.ClosedGroupDetailsSerializer

如何return特定字段queryset DRF

How to return specific field queryset DRF

django

rest

django-queryset

django-views

django-rest-framework