This commit is contained in:
Kyle Fuller 2013-08-02 06:13:48 -07:00
commit 15832d4fdb
2 changed files with 12 additions and 8 deletions

View File

@ -92,13 +92,14 @@ class GenericAPIView(views.APIView):
"""
Return a serializer instance to use with paginated data.
"""
class SerializerClass(self.pagination_serializer_class):
class Meta:
object_serializer_class = self.get_serializer_class()
if self.pagination_serializer_class:
class SerializerClass(self.pagination_serializer_class):
class Meta:
object_serializer_class = self.get_serializer_class()
pagination_serializer_class = SerializerClass
context = self.get_serializer_context()
return pagination_serializer_class(instance=page, context=context)
pagination_serializer_class = SerializerClass
context = self.get_serializer_context()
return pagination_serializer_class(instance=page, context=context)
def paginate_queryset(self, queryset, page_size=None):
"""

View File

@ -90,8 +90,11 @@ class ListModelMixin(object):
page = self.paginate_queryset(self.object_list)
if page is not None:
serializer = self.get_pagination_serializer(page)
else:
serializer = self.get_serializer(self.object_list, many=True)
if serializer:
return Response(serializer.data)
serializer = self.get_serializer(self.object_list, many=True)
return Response(serializer.data)