from rest_framework import serializers

from .models import PlatSimInfo


class PlatSimInfoSerializer(serializers.ModelSerializer):

    def to_representation(self, instance):
        """to_representation自定义序列化数据的返回,针对choice字段"""
        data = super().to_representation(instance)
        data.update(sim_operators=instance.get_sim_operators_display())
        data.update(input_type=instance.get_input_type_display())
        data.update(platform=instance.get_platform_display())
        return data

    class Meta:
        model = PlatSimInfo
        fields = '__all__'
        read_only = ["id", "upl_time"]
        extra_kwargs = {
            "upl_time": {
                "format": "%Y-%m-%d %H:%M:%S"
            }
        }