Django序列化器：创建Dot对象时仅筛选当前用户的标签

你遇到的错误是因为SerializerMethodField和**CurrentUserDefault()**都不能直接用来过滤PrimaryKeyRelatedField的queryset——前者是用于序列化输出的字段，后者是设置默认值的工具，都无法提供过滤queryset所需的实际用户实例。

下面是两种可行的正确方案：

方案1：在序列化器初始化时动态修改queryset #

先给tag字段设置空queryset，然后在__init__方法中获取当前用户并过滤标签：

from rest_framework import serializers
from .models import Dot, Tag

class DotSerializer(serializers.ModelSerializer):
    # 先初始化空queryset，后续动态修改
    tag = serializers.PrimaryKeyRelatedField(queryset=Tag.objects.none())

    class Meta:
        model = Dot
        fields = ['tag', '其他字段']

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        # 从context中获取当前用户（视图需传入context）
        user = self.context.get('user') or self.context.get('request').user
        if user and user.is_authenticated:
            # 假设Tag模型用owner字段关联用户，根据实际字段名调整
            self.fields['tag'].queryset = Tag.objects.filter(owner=user)

如果你的视图没有自动传入用户到context，需要在视图中补充：

from rest_framework.generics import CreateAPIView

class DotCreateView(CreateAPIView):
    serializer_class = DotSerializer
    queryset = Dot.objects.all()

    def get_serializer_context(self):
        context = super().get_serializer_context()
        # 将当前用户传入序列化器context
        context['user'] = self.request.user
        return context

方案2：自定义PrimaryKeyRelatedField子类 #

创建一个自定义字段，在get_queryset方法中直接过滤当前用户的标签：

class UserTagField(serializers.PrimaryKeyRelatedField):
    def get_queryset(self):
        # 从context中获取请求对象，进而拿到用户
        request = self.context.get('request')
        if not request or not request.user.is_authenticated:
            return Tag.objects.none()
        # 根据实际用户关联字段调整过滤条件
        return Tag.objects.filter(owner=request.user)

class DotSerializer(serializers.ModelSerializer):
    tag = UserTagField()

    class Meta:
        model = Dot
        fields = ['tag', '其他字段']

这种方式不需要额外修改视图，因为默认情况下DRF视图会把request传入序列化器的context。

为什么之前的方法失败？ #

• SerializerMethodField：它的作用是在序列化输出时动态生成字段值，无法返回用于过滤queryset的用户实例，所以直接用它过滤会抛出类型错误。
• CurrentUserDefault()：这是一个用于设置字段默认值的工具类（比如自动把当前用户设为某个外键字段的值），它本身不是用户对象，不能直接放到filter的参数中。

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