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

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

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

所以JSON响应的示例如下:

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

Answer 1

你可以使用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')

Answer 2

添加另一种方法 -

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"