如何在Django REST Framework视图中给序列化器传递额外字段？

我已经在序列化器中定义了flag字段，但不知道怎么把视图里的selected_plan值传递过去，让它作为每个返回对象的flag字段值返回。

我的现有代码如下：

serializer.py #

class DataSerializer(serializers.ModelSerializer):

    flag = serializers.BooleanField(required=False)
    class Meta:
        model = Plans
        fields = {
            'id',
            'name',
            'type',
            'details',
            'price'
        }

views.py #

class DetailsView(APIView):
    def get(self, request):
        user_plans = Order.objects.filter(id=request.user).first()
        selected_plan = False
        if user_plans is not None:
            selected_plan = True
            original_plans = Plans.objects.all()
            user_serializer = DataSerializer(original_plans, many=True)
            return Response(user_serializer.data)

期望输出格式：

[
    {
        "id": 1,
        "name": "Name1",
        "type": "normal",
        "details": "NA",
        "price": 2563,
        "flag": true
    },
    {
        "id": 2,
        "name": "Name2",
        "type": "normal",
        "details": "NA",
        "price": 5200,
        "flag": true
    }
]

解决方案 #

方法1：使用SerializerMethodField结合序列化器context传递值 #

这是DRF中传递额外数据的标准方式，灵活性更高。

步骤1：修改序列化器，用SerializerMethodField定义flag并读取context值

class DataSerializer(serializers.ModelSerializer):
    # 替换原BooleanField为SerializerMethodField
    flag = serializers.SerializerMethodField()

    class Meta:
        model = Plans
        fields = ['id', 'name', 'type', 'details', 'price', 'flag']  # 必须把flag加入返回字段列表

    def get_flag(self, obj):
        # 从序列化器的context中获取传入的selected_plan值，默认返回False
        return self.context.get('selected_plan', False)

步骤2：在视图中实例化序列化器时传入context参数

class DetailsView(APIView):
    def get(self, request):
        user_plans = Order.objects.filter(id=request.user).first()
        selected_plan = False
        if user_plans is not None:
            selected_plan = True
        # 修正原代码逻辑：不管user_plans是否存在，都要查询Plans数据
        original_plans = Plans.objects.all()
        # 实例化序列化器时传入context，把selected_plan传递进去
        user_serializer = DataSerializer(original_plans, many=True, context={'selected_plan': selected_plan})
        return Response(user_serializer.data)

方法2：给模型对象动态添加属性（可选） #

如果不想修改序列化器的字段类型，也可以直接给每个Plans对象添加属性，让序列化器自动读取：

步骤1：修改序列化器，确保flag在返回字段中

class DataSerializer(serializers.ModelSerializer):
    flag = serializers.BooleanField(required=False)

    class Meta:
        model = Plans
        fields = ['id', 'name', 'type', 'details', 'price', 'flag']

步骤2：在视图中给每个对象添加flag属性

class DetailsView(APIView):
    def get(self, request):
        user_plans = Order.objects.filter(id=request.user).first()
        selected_plan = False
        if user_plans is not None:
            selected_plan = True
        original_plans = Plans.objects.all()
        # 遍历所有Plans对象，动态添加flag属性
        for plan in original_plans:
            plan.flag = selected_plan
        user_serializer = DataSerializer(original_plans, many=True)
        return Response(user_serializer.data)

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