Change Pagination for collection

This commit is contained in:
Kuroshini 2019-11-18 22:23:23 +03:00
parent 5c4c4c60c3
commit d080f689cb

View File

@ -1,11 +1,10 @@
from django.shortcuts import get_object_or_404 from django.shortcuts import get_object_or_404
from rest_framework import generics from rest_framework import generics, permissions, pagination
from rest_framework import permissions
from collection import models from collection import models
from collection.serializers import common as serializers from collection.serializers import common as serializers
from establishment.serializers import EstablishmentSimilarSerializer from establishment.serializers import EstablishmentSimilarSerializer
from utils.pagination import ProjectPageNumberPagination from utils.pagination import ProjectPageNumberPagination, ProjectMobilePagination
# Mixins # Mixins
@ -52,7 +51,7 @@ class CollectionDetailView(CollectionViewMixin, generics.RetrieveAPIView):
class CollectionEstablishmentListView(CollectionListView): class CollectionEstablishmentListView(CollectionListView):
"""Retrieve list of establishment for collection.""" """Retrieve list of establishment for collection."""
lookup_field = 'slug' lookup_field = 'slug'
pagination_class = ProjectPageNumberPagination pagination_class = ProjectMobilePagination
serializer_class = EstablishmentSimilarSerializer serializer_class = EstablishmentSimilarSerializer
def get_queryset(self): def get_queryset(self):