Added filter class to EstalibshmentListCreateView

This commit is contained in:
evgeniy-st 2019-11-05 16:20:24 +03:00
parent 83bab7f383
commit 568ec84da0

View File

@ -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):