我如何在 django rest 框架中過濾當前用戶的查詢集

Question

class SalonCarDetailsSerializer(serializers.ModelSerializer):

salon = PrimaryKeyRelatedField(queryset=Salon.objects.filter(owner=?))

class Meta:
    model = SalonCarDetails
    fields = ["salon", "car", "price", "number_of_cars"]

CurrentUserDefault() 不起作用

Answer 1

那么，您可以像這樣編寫自己的 PrimaryKeyRelated 字段：

class SalonKeyRelatedField(serializers.PrimaryKeyRelatedField):
    def get_queryset(self):
        qs = super().get_queryset()
        request = self.context.get('request')
        return qs

然后你可以通過 request.user 過濾 qs，這只會在 POST 和 PUT 請求時調用。 然后你可以將它包含在你的序列化器中

salon = SalonKeyRelatedField()

不要忘記在您的領域中包括沙龍

Answer 2

我希望我能看到你的 views.py，但無論如何我對它做了一些假設並將其放在 class 實現場景中。

#
### views.py
#

# Django native libraries
from rest_framework.serializers import Serializer
from rest_framework import viewsets, mixins, status
from django.db.models import Q

# your serializer and model
from .serializers import YourSalonCarSerializer
from .models import SalonCarDetails

class YourCustomViewSet(mixins.RetrieveModelMixin, 
                    mixins.ListModelMixin,
                    mixins.UpdateModelMixin, 
                    mixins.DestroyModelMixin,
                    viewsets.GenericViewSet):

    queryset = SalonCarDetails.objects.all()
    serializer_class = YourSalonCarSerializer

    def get_queryset(self):
        if self.request.user.is_anonymous:
            return []

        lookups = Q(user=self.request.user)
        queryset  = self.queryset.filter(lookups)