gault-millau/apps/establishment/serializers.py
2019-08-27 14:23:11 +03:00

51 lines
1.3 KiB
Python

"""Establishment serializers."""
from rest_framework import serializers
from establishment import models
from location.serializers import AddressSerializer
class EstablishmentTypeSerializer(serializers.ModelSerializer):
"""Serializer for EstablishmentType model."""
class Meta:
"""Meta class."""
model = models.EstablishmentType
fields = ('id', 'name',)
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)
address = AddressSerializer()
class Meta:
"""Meta class."""
model = models.Establishment
fields = (
'id',
'name_trans',
'description_trans',
'public_mark',
'price_level',
'type',
'subtypes',
'image',
'address',
)