diff --git a/apps/establishment/views/back.py b/apps/establishment/views/back.py index e490d576..f6d4d63a 100644 --- a/apps/establishment/views/back.py +++ b/apps/establishment/views/back.py @@ -3,7 +3,7 @@ from django.shortcuts import get_object_or_404 from rest_framework import generics from utils.permissions import IsCountryAdmin, IsEstablishmentManager -from establishment import models, serializers +from establishment import filters, models, serializers from timetable.serialziers import ScheduleRUDSerializer, ScheduleCreateSerializer @@ -17,9 +17,11 @@ class EstablishmentMixinViews: class EstablishmentListCreateView(EstablishmentMixinViews, generics.ListCreateAPIView): """Establishment list/create view.""" + + filter_class = filters.EstablishmentFilter + permission_classes = [IsCountryAdmin | IsEstablishmentManager] queryset = models.Establishment.objects.all() serializer_class = serializers.EstablishmentListCreateSerializer - permission_classes = [IsCountryAdmin|IsEstablishmentManager] class EstablishmentRUDView(generics.RetrieveUpdateDestroyAPIView):