如何使用模型字段过滤 serializers.SlugRelatedField 查询集

How to filter serializers.SlugRelatedField queryset using model field

我无法使用 queryset 参数过滤 SlugRelatedField 的可能选项。这是我的序列化器

class AttendeeProfileSerializer(serializers.HyperlinkedModelSerializer):
    """
    Profile Serializer
    """
    user = serializers.SlugRelatedField(slug_field='username', queryset=User.objects.all())
    module = serializers.SlugRelatedField(slug_field='id', queryset=AttendeeModule.objects.all())
    picture = serializers.ImageField(allow_empty_file=False, required=False)

    class Meta:
        model = AttendeeProfile
        fields = (
            'user', 'module', 'title', 'company', 'email', 'picture', 'is_active', 'created'
        )

虽然我想过滤模块字段,la serializers.SlugRelatedField(slug_field='id', queryset=AttendeeModule.objects.filter(module__app='module__app') 或类似的东西,其中每个模块都属于一个应用程序。

我想我需要通过视图为序列化程序提供更多上下文?我应该覆盖视图上的方法吗?我试过了,但我对 Django 休息框架还是陌生的,它给我带来了很多麻烦

观点是

class AttendeesList(generics.ListCreateAPIView):
    """

    """
    queryset = AttendeeProfile.objects.all()
    serializer_class = AttendeeProfileSerializer

并使用这条路线

url(r'^apps/(?P<url_name>[a-z][a-z0-9]+)/modules/(?P<module_id>[0-9]+)/attendees$',
    views.AttendeesList.as_view(),
    name='attendees-list'),

所以我找到了解决我的古老问题的方法:

class SlugRelatedModuleField(SlugRelatedField):

    def get_queryset(self):
        queryset = self.queryset
        if hasattr(self.root, 'app_id'):
            queryset = queryset.filter(module__app_id=self.root.app_id)
        return queryset

class AttendeeProfileSerializer(ModelSerializer):

    def __init__(self, *args, **kwargs):
        self.app_id = kwargs.pop('app_id')
        super().__init__(*args, **kwargs)

    module = SlugRelatedModuleField(
        slug_field='id', 
        queryset=AttendeeModule.objects.all()
    )

    class Meta:
        model = AttendeeProfile
        fields = ('user', 'module', 'title', 'company', 'email', 'picture', 'is_active', 'created')

这会在 parent/root 序列化器上分配一个 app_id 属性,SlugRelatedModuleField 检查父级以过滤查询集。多田.

在我的例子中,我还需要访问 request.user 以进行过滤,对此我在上面看不到任何解决方案,因此这里有一个示例代码:

class MyFieldName(serializers.SlugRelatedField):
    def get_queryset(self):
        queryset = MyModel.objects.all()
        request = self.context.get('request', None)
        if not request.user.is_superuser:
            queryset = queryset.filter(user=request.user)
        return queryset

class MySerializer(serializers.ModelSerializer):
    organization = MyFieldName(slug_field='slug')

我正在使用序列化程序的 context parameter

class SlugRelatedModuleField(SlugRelatedField):

    def get_queryset(self):
        return self.queryset.filter(module__app_id=self.context["app_id"])

可以像这样在视图(集合)中设置上下文:

class AttendeeProfileViewSet(viewsets.ModelViewSet):
    queryset = AttendeeProfile.objects.all()
    serializer_class = AttendeeProfileSerializer

    def get_serializer_context(self) -> Dict[str, Any]:
        return {
            **super().get_serializer_context(),
            "app_id": self.request.data.get("app_id")
        }