"""Back account serializers"""
from rest_framework import serializers
from partner.models import Partner
from establishment.serializers import EstablishmentShortSerializer
from location.serializers import CountrySimpleSerializer
from location.models import Country


class BackPartnerSerializer(serializers.ModelSerializer):
    establishments = EstablishmentShortSerializer(many=True, read_only=True, source='establishment')
    country = CountrySimpleSerializer(read_only=True)
    type_display = serializers.CharField(read_only=True)
    country_id = serializers.PrimaryKeyRelatedField(
        queryset=Country.objects.all(),
        required=False,
        write_only=True,
        source='country'
    )

    class Meta:
        model = Partner
        fields = (
            'id',
            'name',
            'url',
            # 'image',
            'type',
            'type_display',
            'starting_date',
            'expiry_date',
            'price_per_month',
            'country',
            'country_id',
            'establishments',
        )
        extra_kwargs = {
            'type': {'write_only': True},
        }


class PartnerPicturesSerializer(serializers.ModelSerializer):

    class Meta:
        model = Partner
        fields = (
            'images',
        )