from establishment.serializers import EstablishmentBaseSerializer, PlateSerializer
from rest_framework import serializers
from establishment import models
from main.models import Currency

class EstablishmentListCreateSerializer(EstablishmentBaseSerializer):
    """Establishment create serializer"""

    type_id = serializers.PrimaryKeyRelatedField(
        source='establishment_type',
        queryset=models.EstablishmentType.objects.all(), write_only=True
    )

    class Meta:
        model = models.Establishment
        fields = [
            'id',
            'name',
            'website',
            'phone',
            'email',
            'price_level',
            'toque_number',
            'type_id',
            'type'
        ]


class SocialNetworkSerializers(serializers.ModelSerializer):
    """Social network serializers."""
    class Meta:
        model = models.SocialNetwork
        fields = [
            'id',
            'establishment',
            'title',
            'url',
        ]


class PlatesSerializers(PlateSerializer):
    """Social network serializers."""
    name = serializers.JSONField()
    currency_id = serializers.PrimaryKeyRelatedField(
        source='currency',
        queryset=Currency.objects.all(), write_only=True
    )

    class Meta:
        model = models.Plate
        fields = PlateSerializer.Meta.fields + [
            'name',
            'currency_id',
            'menu'
        ]