102 lines
3.8 KiB
Python
102 lines
3.8 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'))\
|
|
.order_by('locale_name').with_base_related()
|
|
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'))\
|
|
.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]
|
|
|
|
|
|
# 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()
|