66 lines
1.8 KiB
Python
66 lines
1.8 KiB
Python
from location import models
|
|
from location.serializers import common
|
|
from rest_framework import serializers
|
|
from gallery.models import Image
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
|
|
class AddressCreateSerializer(common.AddressDetailSerializer):
|
|
"""Address create serializer."""
|
|
|
|
|
|
class CountryBackSerializer(common.CountrySerializer):
|
|
"""Country back-office serializer."""
|
|
|
|
class Meta:
|
|
model = models.Country
|
|
fields = [
|
|
'id',
|
|
'code',
|
|
'svg_image',
|
|
'name',
|
|
'country_id'
|
|
]
|
|
|
|
|
|
class CityGallerySerializer(serializers.ModelSerializer):
|
|
"""Serializer class for model CityGallery."""
|
|
|
|
class Meta:
|
|
"""Meta class"""
|
|
|
|
model = models.CityGallery
|
|
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."""
|
|
city_pk = self.get_request_kwargs().get('pk')
|
|
image_id = self.get_request_kwargs().get('image_id')
|
|
|
|
city_qs = models.City.objects.filter(pk=city_pk)
|
|
image_qs = Image.objects.filter(id=image_id)
|
|
|
|
if not city_qs.exists():
|
|
raise serializers.ValidationError({'detail': _('City not found')})
|
|
|
|
if not image_qs.exists():
|
|
raise serializers.ValidationError({'detail': _('Image not found')})
|
|
|
|
city = city_qs.first()
|
|
image = image_qs.first()
|
|
|
|
if image in city.gallery.all():
|
|
raise serializers.ValidationError({'detail': _('Image is already added.')})
|
|
|
|
attrs['city'] = city
|
|
attrs['image'] = image
|
|
|
|
return attrs
|