470 lines
18 KiB
Python
470 lines
18 KiB
Python
"""Establishment app views."""
|
|
from collections import defaultdict
|
|
|
|
from django.shortcuts import get_object_or_404
|
|
from django_filters.rest_framework import DjangoFilterBackend
|
|
from rest_framework import generics, permissions
|
|
from rest_framework.response import Response
|
|
|
|
from account.models import User
|
|
from establishment import filters, models, serializers
|
|
from establishment.models import EstablishmentEmployee
|
|
from timetable.models import Timetable
|
|
from timetable.serialziers import ScheduleCreateSerializer, ScheduleRUDSerializer
|
|
from utils.permissions import IsCountryAdmin, IsEstablishmentManager, IsWineryReviewer
|
|
from utils.views import CreateDestroyGalleryViewMixin
|
|
|
|
|
|
class EstablishmentMixinViews:
|
|
"""Establishment mixin."""
|
|
|
|
def get_queryset(self):
|
|
"""Overrided method 'get_queryset'."""
|
|
return models.Establishment.objects.with_base_related()
|
|
|
|
|
|
class EstablishmentListCreateView(EstablishmentMixinViews, generics.ListCreateAPIView):
|
|
"""Establishment list/create view."""
|
|
|
|
filter_class = filters.EstablishmentFilter
|
|
|
|
permission_classes = [IsWineryReviewer | IsCountryAdmin | IsEstablishmentManager]
|
|
queryset = models.Establishment.objects.all()
|
|
serializer_class = serializers.EstablishmentListCreateSerializer
|
|
|
|
|
|
class EmployeeEstablishmentPositionsView(generics.GenericAPIView):
|
|
"""Establishment by employee view."""
|
|
|
|
permission_classes = [IsWineryReviewer | IsCountryAdmin | IsEstablishmentManager]
|
|
queryset = models.EstablishmentEmployee.objects.all()
|
|
serializer_class = serializers.EstablishmentEmployeePositionsSerializer
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
employee_pk = self.kwargs.get('pk')
|
|
|
|
return Response(self.get_serializer(
|
|
self.get_queryset().filter(employee__id=employee_pk).all()
|
|
.prefetch_related('establishment').select_related('position'),
|
|
many=True
|
|
).data)
|
|
|
|
|
|
class EmployeeEstablishmentsListView(generics.ListAPIView):
|
|
"""Establishment by employee list view."""
|
|
|
|
permission_classes = [IsWineryReviewer | IsCountryAdmin | IsEstablishmentManager]
|
|
queryset = models.Establishment.objects.all()
|
|
serializer_class = serializers.EstablishmentListCreateSerializer
|
|
|
|
def get_queryset(self):
|
|
pk = self.kwargs.get('pk')
|
|
employee = get_object_or_404(models.Employee, pk=pk)
|
|
return employee.establishments.with_extended_related()
|
|
|
|
|
|
class EmployeePositionsListView(generics.ListAPIView):
|
|
"""Establishment position by employee list view."""
|
|
|
|
permission_classes = [IsWineryReviewer | IsCountryAdmin | IsEstablishmentManager]
|
|
queryset = models.Establishment.objects.all()
|
|
serializer_class = serializers.EstablishmentPositionListSerializer
|
|
|
|
def get_queryset(self):
|
|
pk = self.kwargs.get('pk')
|
|
employee = get_object_or_404(models.Employee, pk=pk)
|
|
return employee.establishments.with_extended_related()
|
|
|
|
|
|
class EstablishmentRUDView(generics.RetrieveUpdateDestroyAPIView):
|
|
lookup_field = 'slug'
|
|
queryset = models.Establishment.objects.all().prefetch_related(
|
|
'establishmentemployee_set',
|
|
'establishmentemployee_set__establishment',
|
|
)
|
|
serializer_class = serializers.EstablishmentRUDSerializer
|
|
permission_classes = [IsWineryReviewer | IsCountryAdmin | IsEstablishmentManager]
|
|
|
|
|
|
class EstablishmentScheduleRUDView(generics.RetrieveUpdateDestroyAPIView):
|
|
"""Establishment schedule RUD view"""
|
|
lookup_field = 'slug'
|
|
serializer_class = ScheduleRUDSerializer
|
|
permission_classes = [IsWineryReviewer | IsEstablishmentManager]
|
|
|
|
def get_object(self):
|
|
"""
|
|
Returns the object the view is displaying.
|
|
"""
|
|
establishment_slug = self.kwargs['slug']
|
|
schedule_id = self.kwargs['schedule_id']
|
|
|
|
establishment = get_object_or_404(klass=models.Establishment.objects.all(),
|
|
slug=establishment_slug)
|
|
schedule = get_object_or_404(klass=establishment.schedule,
|
|
id=schedule_id)
|
|
|
|
# May raise a permission denied
|
|
self.check_object_permissions(self.request, establishment)
|
|
self.check_object_permissions(self.request, schedule)
|
|
|
|
return schedule
|
|
|
|
|
|
class EstablishmentScheduleCreateView(generics.CreateAPIView):
|
|
"""Establishment schedule Create view"""
|
|
lookup_field = 'slug'
|
|
serializer_class = ScheduleCreateSerializer
|
|
queryset = Timetable.objects.all()
|
|
permission_classes = [IsWineryReviewer | IsEstablishmentManager]
|
|
|
|
|
|
class MenuListCreateView(generics.ListCreateAPIView):
|
|
"""Menu list create view."""
|
|
serializer_class = serializers.MenuSerializers
|
|
queryset = models.Menu.objects.all()
|
|
permission_classes = [IsWineryReviewer | IsEstablishmentManager]
|
|
filter_backends = (DjangoFilterBackend,)
|
|
filterset_fields = (
|
|
'establishment',
|
|
'establishment__slug',
|
|
)
|
|
|
|
|
|
class MenuRUDView(generics.RetrieveUpdateDestroyAPIView):
|
|
"""Menu RUD view."""
|
|
serializer_class = serializers.MenuRUDSerializers
|
|
queryset = models.Menu.objects.all()
|
|
permission_classes = [IsWineryReviewer | IsEstablishmentManager]
|
|
|
|
|
|
class SocialChoiceListCreateView(generics.ListCreateAPIView):
|
|
"""SocialChoice list create view."""
|
|
serializer_class = serializers.SocialChoiceSerializers
|
|
queryset = models.SocialChoice.objects.all()
|
|
pagination_class = None
|
|
permission_classes = [permissions.IsAdminUser]
|
|
|
|
|
|
class SocialChoiceRUDView(generics.RetrieveUpdateDestroyAPIView):
|
|
"""SocialChoice RUD view."""
|
|
serializer_class = serializers.SocialChoiceSerializers
|
|
queryset = models.SocialChoice.objects.all()
|
|
permission_classes = [permissions.IsAdminUser]
|
|
|
|
|
|
class SocialListCreateView(generics.ListCreateAPIView):
|
|
"""Social list create view."""
|
|
serializer_class = serializers.SocialNetworkSerializers
|
|
queryset = models.SocialNetwork.objects.all()
|
|
pagination_class = None
|
|
permission_classes = [permissions.IsAdminUser]
|
|
|
|
|
|
class SocialRUDView(generics.RetrieveUpdateDestroyAPIView):
|
|
"""Social RUD view."""
|
|
serializer_class = serializers.SocialNetworkSerializers
|
|
queryset = models.SocialNetwork.objects.all()
|
|
permission_classes = [IsEstablishmentManager]
|
|
|
|
|
|
class PlateListCreateView(generics.ListCreateAPIView):
|
|
"""Plate list create view."""
|
|
serializer_class = serializers.PlatesSerializers
|
|
queryset = models.Plate.objects.all()
|
|
pagination_class = None
|
|
permission_classes = [IsWineryReviewer | IsEstablishmentManager]
|
|
|
|
|
|
class PlateRUDView(generics.RetrieveUpdateDestroyAPIView):
|
|
"""Plate RUD view."""
|
|
serializer_class = serializers.PlatesSerializers
|
|
queryset = models.Plate.objects.all()
|
|
permission_classes = [IsWineryReviewer | IsEstablishmentManager]
|
|
|
|
|
|
class PhonesListCreateView(generics.ListCreateAPIView):
|
|
"""Phones list create view."""
|
|
serializer_class = serializers.ContactPhoneBackSerializers
|
|
queryset = models.ContactPhone.objects.all()
|
|
pagination_class = None
|
|
permission_classes = [IsWineryReviewer | IsEstablishmentManager]
|
|
|
|
|
|
class PhonesRUDView(generics.RetrieveUpdateDestroyAPIView):
|
|
"""Phones RUD view."""
|
|
serializer_class = serializers.ContactPhoneBackSerializers
|
|
queryset = models.ContactPhone.objects.all()
|
|
permission_classes = [IsWineryReviewer | IsEstablishmentManager]
|
|
|
|
|
|
class EmailListCreateView(generics.ListCreateAPIView):
|
|
"""Email list create view."""
|
|
serializer_class = serializers.ContactEmailBackSerializers
|
|
queryset = models.ContactEmail.objects.all()
|
|
pagination_class = None
|
|
permission_classes = [IsWineryReviewer | IsEstablishmentManager]
|
|
|
|
|
|
class EmailRUDView(generics.RetrieveUpdateDestroyAPIView):
|
|
"""Email RUD view."""
|
|
serializer_class = serializers.ContactEmailBackSerializers
|
|
queryset = models.ContactEmail.objects.all()
|
|
permission_classes = [IsWineryReviewer | IsEstablishmentManager]
|
|
|
|
|
|
class EmployeeListCreateView(generics.ListCreateAPIView):
|
|
"""Emplyoee list create view."""
|
|
permission_classes = (permissions.AllowAny,)
|
|
filter_class = filters.EmployeeBackFilter
|
|
serializer_class = serializers.EmployeeBackSerializers
|
|
queryset = models.Employee.objects.all().distinct().with_back_office_related()
|
|
|
|
|
|
class EmployeesListSearchViews(generics.ListAPIView):
|
|
"""Employee search view"""
|
|
pagination_class = None
|
|
permission_classes = (permissions.AllowAny,)
|
|
queryset = models.Employee.objects.all().with_back_office_related().select_related('photo')
|
|
filter_class = filters.EmployeeBackSearchFilter
|
|
serializer_class = serializers.EmployeeBackSerializers
|
|
|
|
|
|
class EstablishmentEmployeeListView(generics.ListCreateAPIView):
|
|
"""Establishment emplyoees list view."""
|
|
permission_classes = (permissions.AllowAny,)
|
|
serializer_class = serializers.EstEmployeeBackSerializer
|
|
pagination_class = None
|
|
|
|
def get_queryset(self):
|
|
establishment_id = self.kwargs['establishment_id']
|
|
return models.Employee.objects.filter(
|
|
establishmentemployee__establishment_id=establishment_id,
|
|
).distinct()
|
|
|
|
|
|
class EmployeeRUDView(generics.RetrieveUpdateDestroyAPIView):
|
|
"""Employee RUD view."""
|
|
serializer_class = serializers.EmployeeBackSerializers
|
|
queryset = models.Employee.objects.all().with_back_office_related()
|
|
|
|
|
|
class RemoveAwardView(generics.DestroyAPIView):
|
|
lookup_field = 'pk'
|
|
serializer_class = serializers.EmployeeBackSerializers
|
|
queryset = models.Employee.objects.all().with_back_office_related()
|
|
|
|
def get_object(self):
|
|
employee = super().get_object()
|
|
employee.remove_award(self.kwargs['award_id'])
|
|
return employee
|
|
|
|
def delete(self, request, *args, **kwargs):
|
|
instance = self.get_object()
|
|
return Response(status=status.HTTP_204_NO_CONTENT)
|
|
|
|
|
|
class EstablishmentTypeListCreateView(generics.ListCreateAPIView):
|
|
"""Establishment type list/create view."""
|
|
serializer_class = serializers.EstablishmentTypeBaseSerializer
|
|
queryset = models.EstablishmentType.objects.all().select_related('default_image')
|
|
pagination_class = None
|
|
|
|
|
|
class EstablishmentTypeRUDView(generics.RetrieveUpdateDestroyAPIView):
|
|
"""Establishment type retrieve/update/destroy view."""
|
|
serializer_class = serializers.EstablishmentTypeBaseSerializer
|
|
queryset = models.EstablishmentType.objects.all().select_related('default_image')
|
|
|
|
|
|
class EstablishmentSubtypeListCreateView(generics.ListCreateAPIView):
|
|
"""Establishment subtype list/create view."""
|
|
serializer_class = serializers.EstablishmentSubTypeBaseSerializer
|
|
queryset = models.EstablishmentSubType.objects.all().select_related('default_image')
|
|
pagination_class = None
|
|
|
|
|
|
class EstablishmentSubtypeRUDView(generics.RetrieveUpdateDestroyAPIView):
|
|
"""Establishment subtype retrieve/update/destroy view."""
|
|
serializer_class = serializers.EstablishmentSubTypeBaseSerializer
|
|
queryset = models.EstablishmentSubType.objects.all().select_related('default_image')
|
|
|
|
|
|
class EstablishmentGalleryCreateDestroyView(EstablishmentMixinViews,
|
|
CreateDestroyGalleryViewMixin):
|
|
"""Resource for a create|destroy gallery for establishment for back-office users."""
|
|
lookup_field = 'slug'
|
|
serializer_class = serializers.EstablishmentBackOfficeGallerySerializer
|
|
|
|
def get_object(self):
|
|
"""
|
|
Returns the object the view is displaying.
|
|
"""
|
|
establishment_qs = self.filter_queryset(self.get_queryset())
|
|
|
|
establishment = get_object_or_404(establishment_qs, slug=self.kwargs.get('slug'))
|
|
gallery = get_object_or_404(establishment.establishment_gallery,
|
|
image_id=self.kwargs.get('image_id'))
|
|
|
|
# May raise a permission denied
|
|
self.check_object_permissions(self.request, gallery)
|
|
|
|
return gallery
|
|
|
|
|
|
class EstablishmentGalleryListView(EstablishmentMixinViews,
|
|
generics.ListAPIView):
|
|
"""Resource for returning gallery for establishment for back-office users."""
|
|
lookup_field = 'slug'
|
|
serializer_class = serializers.ImageBaseSerializer
|
|
|
|
def get_object(self):
|
|
"""Override get_object method."""
|
|
qs = super(EstablishmentGalleryListView, self).get_queryset()
|
|
establishment = get_object_or_404(qs, slug=self.kwargs.get('slug'))
|
|
|
|
# May raise a permission denied
|
|
self.check_object_permissions(self.request, establishment)
|
|
|
|
return establishment
|
|
|
|
def get_queryset(self):
|
|
"""Override get_queryset method."""
|
|
return self.get_object().crop_gallery
|
|
|
|
|
|
class EstablishmentCompanyListCreateView(EstablishmentMixinViews,
|
|
generics.ListCreateAPIView):
|
|
"""List|Create establishment company view."""
|
|
|
|
lookup_field = 'slug'
|
|
serializer_class = serializers.EstablishmentCompanyListCreateSerializer
|
|
|
|
def get_object(self):
|
|
"""Returns the object the view is displaying."""
|
|
establishment_qs = models.Establishment.objects.all()
|
|
filtered_ad_qs = self.filter_queryset(establishment_qs)
|
|
|
|
establishment = get_object_or_404(filtered_ad_qs, slug=self.kwargs.get('slug'))
|
|
|
|
# May raise a permission denied
|
|
self.check_object_permissions(self.request, establishment)
|
|
|
|
return establishment
|
|
|
|
def get_queryset(self):
|
|
"""Overridden get_queryset method."""
|
|
return self.get_object().companies.all()
|
|
|
|
|
|
class EstablishmentCompanyRUDView(EstablishmentMixinViews,
|
|
generics.RetrieveUpdateDestroyAPIView):
|
|
"""Create|Retrieve|Update|Destroy establishment company view."""
|
|
|
|
lookup_field = 'slug'
|
|
serializer_class = serializers.CompanyBaseSerializer
|
|
|
|
def get_object(self):
|
|
"""Returns the object the view is displaying."""
|
|
establishment_qs = models.Establishment.objects.all()
|
|
filtered_ad_qs = self.filter_queryset(establishment_qs)
|
|
|
|
establishment = get_object_or_404(filtered_ad_qs, slug=self.kwargs.get('slug'))
|
|
company = get_object_or_404(establishment.companies.all(), pk=self.kwargs.get('company_pk'))
|
|
|
|
# May raise a permission denied
|
|
self.check_object_permissions(self.request, company)
|
|
|
|
return company
|
|
|
|
|
|
class EstablishmentNoteListCreateView(EstablishmentMixinViews,
|
|
generics.ListCreateAPIView):
|
|
"""Retrieve|Update|Destroy establishment note view."""
|
|
|
|
lookup_field = 'slug'
|
|
serializer_class = serializers.EstablishmentNoteListCreateSerializer
|
|
|
|
def get_object(self):
|
|
"""Returns the object the view is displaying."""
|
|
establishment_qs = models.Establishment.objects.all()
|
|
filtered_establishment_qs = self.filter_queryset(establishment_qs)
|
|
|
|
establishment = get_object_or_404(filtered_establishment_qs, slug=self.kwargs.get('slug'))
|
|
|
|
# May raise a permission denied
|
|
self.check_object_permissions(self.request, establishment)
|
|
|
|
return establishment
|
|
|
|
def get_queryset(self):
|
|
"""Overridden get_queryset method."""
|
|
return self.get_object().notes.all()
|
|
|
|
|
|
class EstablishmentNoteRUDView(EstablishmentMixinViews,
|
|
generics.RetrieveUpdateDestroyAPIView):
|
|
"""Create|Retrieve|Update|Destroy establishment note view."""
|
|
|
|
lookup_field = 'slug'
|
|
serializer_class = serializers.EstablishmentNoteBaseSerializer
|
|
|
|
def get_object(self):
|
|
"""Returns the object the view is displaying."""
|
|
establishment_qs = models.Establishment.objects.all()
|
|
filtered_establishment_qs = self.filter_queryset(establishment_qs)
|
|
|
|
establishment = get_object_or_404(filtered_establishment_qs, slug=self.kwargs.get('slug'))
|
|
note = get_object_or_404(establishment.notes.all(), pk=self.kwargs['note_pk'])
|
|
|
|
# May raise a permission denied
|
|
self.check_object_permissions(self.request, note)
|
|
|
|
return note
|
|
|
|
|
|
class EstablishmentEmployeeCreateView(generics.CreateAPIView):
|
|
serializer_class = serializers.EstablishmentEmployeeCreateSerializer
|
|
queryset = models.EstablishmentEmployee.objects.all()
|
|
# TODO send email to all admins and add endpoint for changing status
|
|
|
|
|
|
class EstablishmentEmployeeDeleteView(generics.DestroyAPIView):
|
|
queryset = EstablishmentEmployee.objects.all()
|
|
permission_classes = [IsEstablishmentManager | permissions.IsAdminUser]
|
|
|
|
|
|
class EstablishmentPositionListView(generics.ListAPIView):
|
|
"""Establishment positions list view."""
|
|
|
|
pagination_class = None
|
|
permission_classes = (permissions.AllowAny,)
|
|
queryset = models.Position.objects.all()
|
|
serializer_class = serializers.PositionBackSerializer
|
|
|
|
|
|
class EstablishmentAdminView(generics.ListAPIView):
|
|
"""Establishment admin list view."""
|
|
serializer_class = serializers.EstablishmentAdminListSerializer
|
|
permission_classes = (permissions.IsAuthenticatedOrReadOnly,)
|
|
|
|
def get_queryset(self):
|
|
establishment = get_object_or_404(
|
|
models.Establishment, slug=self.kwargs['slug'])
|
|
return User.objects.establishment_admin(establishment).distinct()
|
|
|
|
|
|
class MenuDishesListCreateView(generics.ListCreateAPIView):
|
|
"""Menu (dessert, main_course, starter) list create view."""
|
|
serializer_class = serializers.MenuDishesSerializer
|
|
queryset = models.Menu.objects.with_schedule_plates_establishment().dishes().distinct()
|
|
permission_classes = [IsWineryReviewer | IsEstablishmentManager]
|
|
filter_class = filters.MenuDishesBackFilter
|
|
|
|
|
|
class MenuDishesRUDView(generics.RetrieveUpdateDestroyAPIView):
|
|
"""Menu (dessert, main_course, starter) RUD view."""
|
|
serializer_class = serializers.MenuDishesRUDSerializers
|
|
queryset = models.Menu.objects.dishes().distinct()
|
|
permission_classes = [IsWineryReviewer | IsEstablishmentManager]
|