back menu filter by establishment

apps/establishment/views/back.py

@@ -1,6 +1,7 @@
 """Establishment app views."""
 from django.http import Http404, HttpResponse
 from django.shortcuts import get_object_or_404
+from django_filters.rest_framework import DjangoFilterBackend
 from rest_framework import generics, permissions, status
 
 from establishment import filters, models, serializers
@@ -41,7 +42,7 @@ class EstablishmentScheduleRUDView(generics.RetrieveUpdateDestroyAPIView):
     """Establishment schedule RUD view"""
     lookup_field = 'slug'
     serializer_class = ScheduleRUDSerializer
-    permission_classes = [IsWineryReviewer |IsEstablishmentManager]
+    permission_classes = [IsWineryReviewer | IsEstablishmentManager]
 
     def get_object(self):
         """
@@ -75,6 +76,11 @@ class MenuListCreateView(generics.ListCreateAPIView):
     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):
@@ -161,7 +167,7 @@ class EmailRUDView(generics.RetrieveUpdateDestroyAPIView):
 
 class EmployeeListCreateView(generics.ListCreateAPIView):
     """Emplyoee list create view."""
-    permission_classes = (permissions.AllowAny, )
+    permission_classes = (permissions.AllowAny,)
     filter_class = filters.EmployeeBackFilter
     serializer_class = serializers.EmployeeBackSerializers
     queryset = models.Employee.objects.all()
@@ -170,7 +176,7 @@ class EmployeeListCreateView(generics.ListCreateAPIView):
 
 class EstablishmentEmployeeListView(generics.ListCreateAPIView):
     """Establishment emplyoees list view."""
-    permission_classes = (permissions.AllowAny, )
+    permission_classes = (permissions.AllowAny,)
     serializer_class = serializers.EstablishmentEmployeeBackSerializer
 
     def get_queryset(self):
@@ -352,8 +358,8 @@ class EstablishmentEmployeeCreateView(generics.CreateAPIView):
 class EstablishmentEmployeeDeleteView(generics.DestroyAPIView):
 
     def _get_object_to_delete(self, establishment_id, employee_id):
-        result_qs = models.EstablishmentEmployee\
-            .objects\
+        result_qs = models.EstablishmentEmployee \
+            .objects \
             .filter(establishment_id=establishment_id, employee_id=employee_id)
         if not result_qs.exists():
             raise Http404
@@ -371,6 +377,6 @@ class EstablishmentPositionListView(generics.ListAPIView):
     """Establishment positions list view."""
 
     pagination_class = None
-    permission_classes = (permissions.AllowAny, )
+    permission_classes = (permissions.AllowAny,)
     queryset = models.Position.objects.all()
     serializer_class = serializers.PositionBackSerializer