153 lines
5.7 KiB
Python
153 lines
5.7 KiB
Python
"""Location app views."""
|
|
from rest_framework import generics
|
|
from django.contrib.postgres.fields.jsonb import KeyTextTransform
|
|
from utils.models import get_current_locale
|
|
|
|
from location import models, serializers
|
|
from location.views import common
|
|
from utils.permissions import IsCountryAdmin
|
|
from utils.views import CreateDestroyGalleryViewMixin
|
|
from rest_framework.permissions import IsAuthenticatedOrReadOnly
|
|
from django.shortcuts import get_object_or_404
|
|
from django.db import IntegrityError
|
|
from utils.serializers import ImageBaseSerializer
|
|
from location.filters import RegionFilter
|
|
|
|
from location import filters
|
|
|
|
|
|
# Address
|
|
|
|
|
|
class AddressListCreateView(common.AddressViewMixin, generics.ListCreateAPIView):
|
|
"""Create view for model Address."""
|
|
serializer_class = serializers.AddressDetailSerializer
|
|
queryset = models.Address.objects.all()
|
|
permission_classes = [IsAuthenticatedOrReadOnly | IsCountryAdmin]
|
|
|
|
|
|
class AddressRUDView(common.AddressViewMixin, generics.RetrieveUpdateDestroyAPIView):
|
|
"""RUD view for model Address."""
|
|
serializer_class = serializers.AddressDetailSerializer
|
|
queryset = models.Address.objects.all()
|
|
permission_classes = [IsAuthenticatedOrReadOnly | IsCountryAdmin]
|
|
|
|
|
|
# City
|
|
class CityListCreateView(common.CityViewMixin, generics.ListCreateAPIView):
|
|
"""Create view for model City."""
|
|
serializer_class = serializers.CityBaseSerializer
|
|
permission_classes = [IsAuthenticatedOrReadOnly | IsCountryAdmin]
|
|
queryset = models.City.objects.all()
|
|
filter_class = filters.CityBackFilter
|
|
|
|
def get_queryset(self):
|
|
"""Overridden method 'get_queryset'."""
|
|
qs = models.City.objects.all().annotate(locale_name=KeyTextTransform(get_current_locale(), 'name_translated'))\
|
|
.order_by('locale_name')
|
|
if self.request.country_code:
|
|
qs = qs.by_country_code(self.request.country_code)
|
|
return qs
|
|
|
|
|
|
class CityListSearchView(common.CityViewMixin, generics.ListCreateAPIView):
|
|
"""Create view for model City."""
|
|
serializer_class = serializers.CityBaseSerializer
|
|
permission_classes = [IsAuthenticatedOrReadOnly | IsCountryAdmin]
|
|
queryset = models.City.objects.all()\
|
|
.annotate(locale_name=KeyTextTransform(get_current_locale(), 'name_translated'))\
|
|
.order_by('locale_name')
|
|
filter_class = filters.CityBackFilter
|
|
pagination_class = None
|
|
|
|
|
|
class CityRUDView(common.CityViewMixin, generics.RetrieveUpdateDestroyAPIView):
|
|
"""RUD view for model City."""
|
|
serializer_class = serializers.CityDetailSerializer
|
|
permission_classes = [IsAuthenticatedOrReadOnly | IsCountryAdmin]
|
|
|
|
|
|
class CityGalleryCreateDestroyView(common.CityViewMixin,
|
|
CreateDestroyGalleryViewMixin):
|
|
"""Resource for a create gallery for product for back-office users."""
|
|
serializer_class = serializers.CityGallerySerializer
|
|
permission_classes = [IsAuthenticatedOrReadOnly | IsCountryAdmin]
|
|
|
|
def get_object(self):
|
|
"""
|
|
Returns the object the view is displaying.
|
|
"""
|
|
city_qs = self.filter_queryset(self.get_queryset())
|
|
|
|
city = get_object_or_404(city_qs, pk=self.kwargs.get('pk'))
|
|
gallery = get_object_or_404(city.city_gallery, image_id=self.kwargs.get('image_id'))
|
|
|
|
# May raise a permission denied
|
|
self.check_object_permissions(self.request, gallery)
|
|
|
|
return gallery
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
try:
|
|
return super(CityGalleryCreateDestroyView, self).create(request, *args, **kwargs)
|
|
except IntegrityError as e:
|
|
if not 'unique constraint' in e.args[0]:
|
|
raise e
|
|
models.CityGallery.objects.filter(city=kwargs['pk'], is_main=request.data['is_main']).delete()
|
|
return super(CityGalleryCreateDestroyView, self).create(request, *args, **kwargs)
|
|
|
|
|
|
class CityGalleryListView(common.CityViewMixin,
|
|
generics.ListAPIView):
|
|
"""Resource for returning gallery for product for back-office users."""
|
|
serializer_class = ImageBaseSerializer
|
|
permission_classes = [IsAuthenticatedOrReadOnly | IsCountryAdmin]
|
|
|
|
def get_object(self):
|
|
"""Override get_object method."""
|
|
qs = super(CityGalleryListView, self).get_queryset()
|
|
city = get_object_or_404(qs, pk=self.kwargs['pk'])
|
|
|
|
# May raise a permission denied
|
|
self.check_object_permissions(self.request, city)
|
|
|
|
return city
|
|
|
|
def get_queryset(self):
|
|
"""Override get_queryset method."""
|
|
return self.get_object().crop_gallery
|
|
|
|
|
|
# Region
|
|
class RegionListCreateView(common.RegionViewMixin, generics.ListCreateAPIView):
|
|
"""Create view for model Region"""
|
|
pagination_class = None
|
|
serializer_class = serializers.RegionSerializer
|
|
permission_classes = [IsAuthenticatedOrReadOnly | IsCountryAdmin]
|
|
# ordering_fields = 'name'
|
|
filter_class = RegionFilter
|
|
|
|
|
|
class RegionRUDView(common.RegionViewMixin, generics.RetrieveUpdateDestroyAPIView):
|
|
"""Retrieve view for model Region"""
|
|
serializer_class = serializers.RegionSerializer
|
|
permission_classes = [IsAuthenticatedOrReadOnly | IsCountryAdmin]
|
|
|
|
|
|
# Country
|
|
class CountryListCreateView(generics.ListCreateAPIView):
|
|
"""List/Create view for model Country."""
|
|
queryset = models.Country.objects.all()\
|
|
.annotate(locale_name=KeyTextTransform(get_current_locale(), 'name'))\
|
|
.order_by('locale_name')
|
|
serializer_class = serializers.CountryBackSerializer
|
|
pagination_class = None
|
|
permission_classes = [IsAuthenticatedOrReadOnly | IsCountryAdmin]
|
|
|
|
|
|
class CountryRUDView(generics.RetrieveUpdateDestroyAPIView):
|
|
"""RUD view for model Country."""
|
|
serializer_class = serializers.CountryBackSerializer
|
|
permission_classes = [IsAuthenticatedOrReadOnly | IsCountryAdmin]
|
|
queryset = models.Country.objects.all()
|