diff --git a/weni/internal/flows/views.py b/weni/internal/flows/views.py index 84bfa0199..ae4c819b3 100644 --- a/weni/internal/flows/views.py +++ b/weni/internal/flows/views.py @@ -8,9 +8,14 @@ from temba.flows.models import Flow +class FlowPagination(PageNumberPagination): + page_size_query_param = 'page_size' + max_page_size = 20 + + class FlowViewSet(CreateModelMixin, InternalGenericViewSet, ListModelMixin): serializer_class = FlowSerializer - pagination_class = PageNumberPagination + pagination_class = FlowPagination def get_queryset(self): serializer = self.get_serializer(data=self.request.query_params.dict()) @@ -22,7 +27,7 @@ def get_queryset(self): ).exclude(is_archived=True) if queryset: - return self.paginate_queryset(queryset) + return queryset raise NotFound()