280 lines
8.3 KiB
Python
280 lines
8.3 KiB
Python
from rest_framework import serializers
|
|
|
|
from establishment import models
|
|
from establishment import serializers as model_serializers
|
|
from location.serializers import AddressDetailSerializer
|
|
from main.models import Currency
|
|
from utils.decorators import with_base_attributes
|
|
from utils.serializers import TimeZoneChoiceField
|
|
from gallery.models import Image
|
|
from django.utils.translation import gettext_lazy as _
|
|
from account.serializers.common import UserShortSerializer
|
|
|
|
|
|
class EstablishmentListCreateSerializer(model_serializers.EstablishmentBaseSerializer):
|
|
"""Establishment create serializer"""
|
|
|
|
type_id = serializers.PrimaryKeyRelatedField(
|
|
source='establishment_type',
|
|
queryset=models.EstablishmentType.objects.all(),
|
|
write_only=True
|
|
)
|
|
phones = model_serializers.ContactPhonesSerializer(read_only=True,
|
|
many=True, )
|
|
emails = model_serializers.ContactEmailsSerializer(read_only=True,
|
|
many=True, )
|
|
socials = model_serializers.SocialNetworkRelatedSerializers(read_only=True,
|
|
many=True, )
|
|
type = model_serializers.EstablishmentTypeBaseSerializer(source='establishment_type',
|
|
read_only=True)
|
|
tz = TimeZoneChoiceField()
|
|
|
|
class Meta:
|
|
model = models.Establishment
|
|
fields = [
|
|
'id',
|
|
'name',
|
|
'transliterated_name',
|
|
'index_name',
|
|
'website',
|
|
'phones',
|
|
'emails',
|
|
'price_level',
|
|
'toque_number',
|
|
'type_id',
|
|
'type',
|
|
'socials',
|
|
'image_url',
|
|
'slug',
|
|
# TODO: check in admin filters
|
|
'is_publish',
|
|
'guestonline_id',
|
|
'lastable_id',
|
|
'tags',
|
|
'tz',
|
|
]
|
|
|
|
|
|
class EstablishmentRUDSerializer(model_serializers.EstablishmentBaseSerializer):
|
|
"""Establishment create serializer"""
|
|
|
|
type_id = serializers.PrimaryKeyRelatedField(
|
|
source='establishment_type',
|
|
queryset=models.EstablishmentType.objects.all(), write_only=True
|
|
)
|
|
address = AddressDetailSerializer()
|
|
phones = model_serializers.ContactPhonesSerializer(read_only=False,
|
|
many=True, )
|
|
emails = model_serializers.ContactEmailsSerializer(read_only=False,
|
|
many=True, )
|
|
socials = model_serializers.SocialNetworkRelatedSerializers(read_only=False,
|
|
many=True, )
|
|
type = model_serializers.EstablishmentTypeBaseSerializer(source='establishment_type')
|
|
|
|
class Meta:
|
|
model = models.Establishment
|
|
fields = [
|
|
'id',
|
|
'name',
|
|
'website',
|
|
'phones',
|
|
'emails',
|
|
'price_level',
|
|
'toque_number',
|
|
'type_id',
|
|
'type',
|
|
'socials',
|
|
'image_url',
|
|
# TODO: check in admin filters
|
|
'is_publish',
|
|
'address',
|
|
'tags',
|
|
]
|
|
|
|
|
|
class SocialChoiceSerializers(serializers.ModelSerializer):
|
|
"""SocialChoice serializers."""
|
|
|
|
class Meta:
|
|
model = models.SocialChoice
|
|
fields = '__all__'
|
|
|
|
|
|
class SocialNetworkSerializers(serializers.ModelSerializer):
|
|
"""Social network serializers."""
|
|
|
|
class Meta:
|
|
model = models.SocialNetwork
|
|
fields = [
|
|
'id',
|
|
'establishment',
|
|
'network',
|
|
'url',
|
|
]
|
|
|
|
|
|
class PlatesSerializers(model_serializers.PlateSerializer):
|
|
"""Plates serializers."""
|
|
|
|
currency_id = serializers.PrimaryKeyRelatedField(
|
|
source='currency',
|
|
queryset=Currency.objects.all(), write_only=True
|
|
)
|
|
|
|
class Meta:
|
|
"""Meta class."""
|
|
|
|
model = models.Plate
|
|
fields = model_serializers.PlateSerializer.Meta.fields + [
|
|
'name',
|
|
'currency_id',
|
|
'menu'
|
|
]
|
|
|
|
|
|
class ContactPhoneBackSerializers(model_serializers.PlateSerializer):
|
|
"""ContactPhone serializers."""
|
|
|
|
class Meta:
|
|
model = models.ContactPhone
|
|
fields = [
|
|
'id',
|
|
'establishment',
|
|
'phone'
|
|
]
|
|
|
|
|
|
class ContactEmailBackSerializers(model_serializers.PlateSerializer):
|
|
"""ContactEmail serializers."""
|
|
|
|
class Meta:
|
|
model = models.ContactEmail
|
|
fields = [
|
|
'id',
|
|
'establishment',
|
|
'email'
|
|
]
|
|
|
|
|
|
# TODO: test decorator
|
|
@with_base_attributes
|
|
class EmployeeBackSerializers(serializers.ModelSerializer):
|
|
"""Employee serializers."""
|
|
|
|
class Meta:
|
|
model = models.Employee
|
|
fields = [
|
|
'id',
|
|
'user',
|
|
'name'
|
|
]
|
|
|
|
|
|
class EstablishmentBackOfficeGallerySerializer(serializers.ModelSerializer):
|
|
"""Serializer class for model EstablishmentGallery."""
|
|
|
|
class Meta:
|
|
"""Meta class"""
|
|
|
|
model = models.EstablishmentGallery
|
|
fields = [
|
|
'id',
|
|
'is_main',
|
|
]
|
|
|
|
def get_request_kwargs(self):
|
|
"""Get url kwargs from request."""
|
|
return self.context.get('request').parser_context.get('kwargs')
|
|
|
|
def validate(self, attrs):
|
|
"""Override validate method."""
|
|
establishment_pk = self.get_request_kwargs().get('pk')
|
|
image_id = self.get_request_kwargs().get('image_id')
|
|
|
|
establishment_qs = models.Establishment.objects.filter(pk=establishment_pk)
|
|
image_qs = Image.objects.filter(id=image_id)
|
|
|
|
if not establishment_qs.exists():
|
|
raise serializers.ValidationError({'detail': _('Establishment not found')})
|
|
|
|
if not image_qs.exists():
|
|
raise serializers.ValidationError({'detail': _('Image not found')})
|
|
|
|
establishment = establishment_qs.first()
|
|
image = image_qs.first()
|
|
|
|
if image in establishment.gallery.all():
|
|
raise serializers.ValidationError({'detail': _('Image is already added.')})
|
|
|
|
attrs['establishment'] = establishment
|
|
attrs['image'] = image
|
|
|
|
return attrs
|
|
|
|
|
|
class EstablishmentCompanyListCreateSerializer(model_serializers.CompanyBaseSerializer):
|
|
"""Serializer for linking page w/ advertisement."""
|
|
|
|
class Meta(model_serializers.CompanyBaseSerializer.Meta):
|
|
"""Meta class."""
|
|
model_serializers.CompanyBaseSerializer.Meta.extra_kwargs.update({
|
|
'establishment': {'required': False}
|
|
})
|
|
|
|
def create(self, validated_data):
|
|
"""Overridden create method."""
|
|
validated_data['establishment'] = self.context.get('view').get_object()
|
|
return super().create(validated_data)
|
|
|
|
|
|
class EstablishmentNoteBaseSerializer(serializers.ModelSerializer):
|
|
"""Serializer for model EstablishmentNote."""
|
|
|
|
user_detail = UserShortSerializer(read_only=True, source='user')
|
|
|
|
class Meta:
|
|
"""Meta class."""
|
|
model = models.EstablishmentNote
|
|
fields = [
|
|
'id',
|
|
'created',
|
|
'modified',
|
|
'text',
|
|
'user',
|
|
'user_detail',
|
|
'establishment',
|
|
]
|
|
extra_kwargs = {
|
|
'created': {'read_only': True},
|
|
'modified': {'read_only': True},
|
|
'establishment': {'required': False, 'write_only': True},
|
|
'user': {'required': False, 'write_only': True},
|
|
}
|
|
|
|
@property
|
|
def serializer_view(self):
|
|
"""Return view instance."""
|
|
return self.context.get('view')
|
|
|
|
|
|
class EstablishmentNoteListCreateSerializer(EstablishmentNoteBaseSerializer):
|
|
"""Serializer for List|Create action for model EstablishmentNote."""
|
|
|
|
def create(self, validated_data):
|
|
"""Overridden create method."""
|
|
validated_data['user'] = self.user
|
|
validated_data['establishment'] = self.establishment
|
|
return super().create(validated_data)
|
|
|
|
@property
|
|
def user(self):
|
|
"""Return user instance from view."""
|
|
if self.serializer_view:
|
|
return self.serializer_view.request.user
|
|
|
|
@property
|
|
def establishment(self):
|
|
"""Return establishment instance from view."""
|
|
if self.serializer_view:
|
|
return self.serializer_view.get_object()
|