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