django restframework choice 自定义输出数据

我有一个这样的需求，返回的数据json中返回的是id，但是我想要得到该id对应的name。

id对应的name

PlatformType = (   
    (0, '通用'),   
    (1, '前装'),   
    (2, '后装'),   
    (3, '海外前装'),   
    (4, '海外后装'),   
    (5, '小系统')
)

class TrackSerializer(serializers.ModelSerializer):
    
    platform = serializers.ChoiceField(choices=PlatformType)
    
    class Meta:    
        model = Track    
        fields = "__all__"

返回的结果是:

{
    platform: 1
}

但是我想要的是1对应的前装，这个时候需要自定义返回的数据。

解决：

1. 自定义字段类型，重写ChoiceField字段类，并重写to_representation方法，在序列化platform字段时，会调用to_representation方法转换成我们想要的格式。

class PlatFormField(serializers.ChoiceField):    
    def to_representation(self, value: Any):        
        return self.choices[value]

class TrackSerializer(serializers.ModelSerializer):
    
    platform = PlatFormField(choices=PlatformType)
    
    class Meta:    
        model = Track    
        fields = "__all__"

2. 重写显示的字段。

将platform字段重新进行改写，获取其显示的名字。

class TrackSerializer(serializers.ModelSerializer):
    platform = serializers.SerializerMethodField()
    class Meta:
        model = Track
        fields = "__all__"

def get_platform(self, obj):
    return obj.get_platform_display()

欢迎关注，互相学习，共同进步~

我的个人博客

我的微信公众号：编程黑洞