from rest_framework import serializers
from .models import Company, Consumer


class CompanySerializer(serializers.ModelSerializer):
    """
    公司序列化
    """
    create_time = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S", required=False, read_only=True)
    update_time = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S", required=False, read_only=True)
    class Meta:
        model = Company
        fields = '__all__'


class ConsumerSerializer(serializers.ModelSerializer):
    """
    客户序列化
    """
    create_time = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S", required=False, read_only=True)
    update_time = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S", required=False, read_only=True)
    company_name = serializers.StringRelatedField(source='company')

    class Meta:
        model = Consumer
        fields = '__all__'
        depth = 1

class ConsumerCUSerializer(serializers.ModelSerializer):
    """
    客户新增编辑序列化
    """
    create_time = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S", required=False, read_only=True)
    update_time = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S", required=False, read_only=True)

    class Meta:
        model = Consumer
        fields = '__all__'