如何在Django的CreateAPIView中动态过滤序列化器查询集？

要在运行时基于请求数据过滤序列化器内的查询集，核心是利用序列化器的context传递请求对象，然后在序列化器初始化时动态修改字段的queryset属性。以下是具体实现步骤：

1. 修改序列化器代码 #

首先，将PrimaryKeyRelatedField的静态queryset设为None，然后在序列化器的__init__方法中，从上下文获取请求并动态生成过滤后的查询集：

from django.contrib.auth import get_user_model
from rest_framework import serializers

class AssignWorkItemSerializer(serializers.Serializer, WorkitemSerializerMixin):
    assigneeIds = serializers.ListSerializer(
        child=serializers.PrimaryKeyRelatedField(queryset=None),
        source='assignees'
    )

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        # 从序列化器上下文获取请求对象
        request = self.context.get('request')
        if request:
            # 这里编写你的动态过滤逻辑，示例中假设根据当前用户的团队过滤分析师
            # 可根据实际需求替换为request.GET/request.data中的参数或其他请求数据
            filtered_analysts = get_user_model().objects.filter(
                userrole__role__name='analyst',
                team=request.user.team  # 示例过滤条件，按需修改
            )
            # 为字段设置动态生成的查询集
            self.fields['assigneeIds'].child.queryset = filtered_analysts

2. 确保视图传递请求上下文 #

由于你的视图基于CreateAPIView，它继承自GenericAPIView，默认会自动将请求对象传入序列化器的context中，因此无需额外配置。如果是自定义视图，需确保在实例化序列化器时传入context={'request': request}。

关于你提到的"查询返回多行数据导致失败"的说明 #

PrimaryKeyRelatedField本身支持多行结果的查询集，报错的真实原因大概率是传入的assigneeIds包含不在查询集内的ID，或是序列化器验证逻辑的其他问题。通过动态过滤查询集，确保传入的ID属于允许的范围，即可解决这类验证错误。

内容的提问来源于stack exchange，提问作者Larry Martell