55 lines
1.4 KiB
Python
55 lines
1.4 KiB
Python
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'
|
|
] |