diff --git a/apps/main/views/back.py b/apps/main/views/back.py index e819b71d..b17a692c 100644 --- a/apps/main/views/back.py +++ b/apps/main/views/back.py @@ -8,7 +8,7 @@ from rest_framework.response import Response from main import serializers from main import tasks from main.filters import AwardFilter -from main.models import Award, Footer, PageType, Panel +from main.models import Award, Footer, PageType, Panel, SiteFeature, Feature from main.views import SiteSettingsView, SiteListView @@ -46,21 +46,29 @@ class ContentTypeView(generics.ListAPIView): class FeatureBackView(generics.ListCreateAPIView): """Feature list or create View.""" serializer_class = serializers.FeatureSerializer + queryset = Feature.objects.all() class SiteFeatureBackView(generics.ListCreateAPIView): """Feature list or create View.""" serializer_class = serializers.SiteFeatureSerializer + queryset = SiteFeature.objects.all() + pagination_class = None + permission_classes = [permissions.IsAdminUser] class FeatureRUDBackView(generics.RetrieveUpdateDestroyAPIView): """Feature RUD View.""" serializer_class = serializers.FeatureSerializer + queryset = SiteFeature.objects.all() + permission_classes = [permissions.IsAdminUser] class SiteFeatureRUDBackView(generics.RetrieveUpdateDestroyAPIView): """Feature RUD View.""" serializer_class = serializers.SiteFeatureSerializer + queryset = SiteFeature.objects.all() + permission_classes = [permissions.IsAdminUser] class SiteSettingsBackOfficeView(SiteSettingsView):