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


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

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

        model = models.EstablishmentType
        fields = ('id', 'name', 'use_subtypes')


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

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

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


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

    name_trans = serializers.CharField()
    description_trans = serializers.CharField()
    type = EstablishmentTypeSerializer(source='establishment_type')
    subtypes = EstablishmentSubTypeSerializer(many=True)

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

        model = models.Establishment
        fields = (
            'id',
            'name_trans',
            'description_trans',
            'public_mark',
            'type',
            'subtypes',
            'address',
        )