116 lines
3.9 KiB
Python
116 lines
3.9 KiB
Python
"""Establishment app views."""
|
|
|
|
from rest_framework import generics
|
|
|
|
from establishment import models
|
|
from establishment import serializers
|
|
from utils.permissions import IsCountryAdmin, IsEstablishmentManager
|
|
|
|
|
|
class EstablishmentMixinViews:
|
|
"""Establishment mixin."""
|
|
|
|
def get_queryset(self):
|
|
"""Overrided method 'get_queryset'."""
|
|
return models.Establishment.objects.published().with_base_related()
|
|
|
|
|
|
class EstablishmentListCreateView(EstablishmentMixinViews, generics.ListCreateAPIView):
|
|
"""Establishment list/create view."""
|
|
queryset = models.Establishment.objects.all()
|
|
serializer_class = serializers.EstablishmentListCreateSerializer
|
|
permission_classes = [IsCountryAdmin|IsEstablishmentManager]
|
|
|
|
|
|
class EstablishmentRUDView(generics.RetrieveUpdateDestroyAPIView):
|
|
queryset = models.Establishment.objects.all()
|
|
serializer_class = serializers.EstablishmentRUDSerializer
|
|
permission_classes = [IsCountryAdmin|IsEstablishmentManager]
|
|
|
|
|
|
class MenuListCreateView(generics.ListCreateAPIView):
|
|
"""Menu list create view."""
|
|
serializer_class = serializers.MenuSerializers
|
|
queryset = models.Menu.objects.all()
|
|
permission_classes = [IsEstablishmentManager]
|
|
|
|
|
|
class MenuRUDView(generics.RetrieveUpdateDestroyAPIView):
|
|
"""Menu RUD view."""
|
|
serializer_class = serializers.MenuRUDSerializers
|
|
queryset = models.Menu.objects.all()
|
|
permission_classes = [IsEstablishmentManager]
|
|
|
|
|
|
class SocialListCreateView(generics.ListCreateAPIView):
|
|
"""Social list create view."""
|
|
serializer_class = serializers.SocialNetworkSerializers
|
|
queryset = models.SocialNetwork.objects.all()
|
|
pagination_class = None
|
|
permission_classes = [IsEstablishmentManager]
|
|
|
|
|
|
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 = [IsEstablishmentManager]
|
|
|
|
|
|
class PlateRUDView(generics.RetrieveUpdateDestroyAPIView):
|
|
"""Social RUD view."""
|
|
serializer_class = serializers.PlatesSerializers
|
|
queryset = models.Plate.objects.all()
|
|
permission_classes = [IsEstablishmentManager]
|
|
|
|
|
|
class PhonesListCreateView(generics.ListCreateAPIView):
|
|
"""Plate list create view."""
|
|
serializer_class = serializers.ContactPhoneBackSerializers
|
|
queryset = models.ContactPhone.objects.all()
|
|
pagination_class = None
|
|
permission_classes = [IsEstablishmentManager]
|
|
|
|
|
|
class PhonesRUDView(generics.RetrieveUpdateDestroyAPIView):
|
|
"""Social RUD view."""
|
|
serializer_class = serializers.ContactPhoneBackSerializers
|
|
queryset = models.ContactPhone.objects.all()
|
|
permission_classes = [IsEstablishmentManager]
|
|
|
|
|
|
class EmailListCreateView(generics.ListCreateAPIView):
|
|
"""Plate list create view."""
|
|
serializer_class = serializers.ContactEmailBackSerializers
|
|
queryset = models.ContactEmail.objects.all()
|
|
pagination_class = None
|
|
permission_classes = [IsEstablishmentManager]
|
|
|
|
|
|
class EmailRUDView(generics.RetrieveUpdateDestroyAPIView):
|
|
"""Social RUD view."""
|
|
serializer_class = serializers.ContactEmailBackSerializers
|
|
queryset = models.ContactEmail.objects.all()
|
|
permission_classes = [IsEstablishmentManager]
|
|
|
|
|
|
class EmployeeListCreateView(generics.ListCreateAPIView):
|
|
"""Emplyoee list create view."""
|
|
serializer_class = serializers.EmployeeBackSerializers
|
|
queryset = models.Employee.objects.all()
|
|
pagination_class = None
|
|
|
|
|
|
class EmployeeRUDView(generics.RetrieveUpdateDestroyAPIView):
|
|
"""Social RUD view."""
|
|
serializer_class = serializers.EmployeeBackSerializers
|
|
queryset = models.Employee.objects.all()
|