from rest_framework import serializers
import datetime

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

    def update(self, instance, validated_data):
        validated_data["upl_time"] = datetime.datetime.now()
        return super().update(instance, validated_data)

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