47 lines
1.2 KiB
Python
47 lines
1.2 KiB
Python
"""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',
|
|
)
|