from rest_framework import serializers

from apps.core.models import *


class AiAlgorithmClassSerializer(serializers.ModelSerializer):
    class Meta:
        model = AiAlgorithmClass
        fields = "__all__"


class AiAlgorithmSerializer(serializers.ModelSerializer):
    create_time = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S")

    class Meta:
        model = AiAlgorithm
        fields = "__all__"


class AiAlgorithmEngineSerializer(serializers.ModelSerializer):
    class Meta:
        model = AiAlgorithmEngine
        fields = "__all__"


class AiAlgorithmParamsSerializer(serializers.ModelSerializer):
    class Meta:
        model = AiAlgorithmParams
        fields = "__all__"


class AiAlgorithmOutputSerializer(serializers.ModelSerializer):
    class Meta:
        model = AiAlgorithmOutput
        fields = "__all__"


class AiAlgorithmModelsSerializer(serializers.ModelSerializer):
    upload_time = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S")
    model_createtime = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S")

    class Meta:
        model = AiAlgorithmModels
        fields = "__all__"


class AiDatasetSerializer(serializers.ModelSerializer):
    create_time = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S")

    class Meta:
        model = AiDataset
        fields = "__all__"


class AiDatasetSampleSerializer(serializers.ModelSerializer):
    create_time = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S")
    annotation_content = serializers.SerializerMethodField()

    class Meta:
        model = AiDatasetSample
        fields = "__all__"

    def get_annotation_content(self, obj):
        return obj.polygon or obj.rectangle or ""


class AIAnnotateTeamSerializer(serializers.ModelSerializer):
    class Meta:
        model = AIAnnotateTeam
        fields = "__all__"


class AIAnnotateTeamMemberSerializer(serializers.ModelSerializer):
    class Meta:
        model = AIAnnotateTeamMember
        fields = "__all__"


class AiAnnotateTaskSerializer(serializers.ModelSerializer):
    create_time = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S")

    class Meta:
        model = AiAnnotateTask
        fields = "__all__"


class AiAnnotateTaskSampleSerializer(serializers.ModelSerializer):
    class Meta:
        model = AiAnnotateTaskSample
        fields = "__all__"


class AiTempFileSerializer(serializers.ModelSerializer):
    create_time = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S")

    class Meta:
        model = AiTempFile
        fields = "__all__"


class AiAlgorithmTrainRecordsSerializer(serializers.ModelSerializer):
    create_time = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S")

    class Meta:
        model = AiAlgorithmTrainRecords
        fields = "__all__"


class AiAlgorithmCallLogSerializer(serializers.ModelSerializer):
    log_time = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S")

    class Meta:
        model = AiAlgorithmCallLog
        fields = "__all__"


class AuthOrganizationSerializer(serializers.ModelSerializer):
    class Meta:
        model = AuthOrganization
        fields = "__all__"


class AuthPersonSerializer(serializers.ModelSerializer):
    date_joined = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S")

    class Meta:
        model = AuthPerson
        fields = "__all__"


class AuthGroupSerializer(serializers.ModelSerializer):
    class Meta:
        model = AuthGroup
        fields = "__all__"


class AIReliabilityObjectsSerializer(serializers.ModelSerializer):
    create_time = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S")

    class Meta:
        model = AIReliabilityObjects
        fields = "__all__"


class AIReliabilityObjectClassSerializer(serializers.ModelSerializer):
    create_time = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S")

    class Meta:
        model = AIReliabilityObjectClass
        fields = "__all__"


class AIReliabilityPlantSerializer(serializers.ModelSerializer):
    class Meta:
        model = AIReliabilityPlant
        fields = "__all__"


class AiSmartTerminalSerializer(serializers.ModelSerializer):
    record_time = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S", read_only=True)
    model_publish_time = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S", read_only=True)
    edge_update_time = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S", read_only=True)
    last_heartbeat_time = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S", read_only=True)
    dtin_tm = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S", read_only=True)

    class Meta:
        model = AiSmartTerminal
        fields = "__all__"