Django REST——如何在返回 REST 响应之前“修改”值？

Question

我做了一些研究，但我不太确定我在技术上寻找什么，所以我没有找到太多信息。我正在使用 Django REST。

型号

class Car(models.Model)
    name =  name = models.CharField(blank=True, null=True, db_index=True)
    speed = models.IntegerField(default=SPEED.FAST)

查看

class CarViewSet(viewsets.ModelViewSet):
    queryset = Car.objects.all()
    serializer_class = CarSerializer

序列化器

class CarSerializer(serializers.ModelSerializer):
    class Meta:
        model = Car

问题：我在Car 模型中的speed 字段是一个整数，当我访问REST 端点API 时，我不想返回一个整数，而是一个字符串。基本上：

if speed is 0:
    return "slow"
else:
    return "fast" 

所以一个示例 JSON 响应是：

{ name: "ferrari", speed: "fast" }

Answer 1

添加另一种方法 -

class CarSerializer(serializers.ModelSerializer):

    class Meta:
        model = Car
        fields = ('name', 'speed')

    def to_representation(self, data):
        data = super(CarSerializer, self).to_representation(data)
        data['speed'] = 'slow' if data.get('speed') == 0 else 'fast'
        return data

此外，@Gocht 答案的一项改进。 更好的做法是在方法 get_speed 中使用条件返回时省略额外的 else 语句 -

def get_speed(self, obj):
    if obj.speed == 0:
        return "slow"
    return "fast"

Answer 2

你可以使用SerializerMethodField:

class CarSerializer(serializers.ModelSerializer):

    speed = serializers.SerializerMethodField()

    def get_speed(self, obj):
        if obj.speed == 0:
            return "slow"
        else:
            return "fast"

    class Meta:
        model = Car
        fields = ('name', 'speed')