"""Establishment serializers."""
from rest_framework import serializers
from establishment import models
from location.serializers import AddressSerializer


class EstablishmentTypeSerializer(serializers.ModelSerializer):
    """Serializer for EstablishmentType model."""

    name_translated = serializers.CharField(allow_null=True)

    class Meta:
        """Meta class."""

        model = models.EstablishmentType
        fields = ('id', 'name_translated')


class EstablishmentSubTypeSerializer(serializers.ModelSerializer):
    """Serializer for EstablishmentSubType models."""

    name_translated = serializers.CharField(allow_null=True)

    class Meta:
        """Meta class."""

        model = models.EstablishmentSubType
        fields = ('id', 'name_translated')


class EstablishmentSerializer(serializers.ModelSerializer):
    """Serializer for Establishment model."""

    name_translated = serializers.CharField(allow_null=True)
    description_translated = serializers.CharField(allow_null=True)
    type = EstablishmentTypeSerializer(source='establishment_type')
    subtypes = EstablishmentSubTypeSerializer(many=True)
    address = AddressSerializer()

    class Meta:
        """Meta class."""

        model = models.Establishment
        fields = (
            'id',
            'name_translated',
            'description_translated',
            'public_mark',
            'price_level',
            'type',
            'subtypes',
            'image',
            'address',
        )