diff options
| author | Tom Christie | 2012-10-01 15:49:19 +0100 | 
|---|---|---|
| committer | Tom Christie | 2012-10-01 15:49:19 +0100 | 
| commit | b16fb5777168246b1e217640b818a82eb6e2141b (patch) | |
| tree | d45a059c40ee2be8c77f6da0c7092a1a57a00ad1 /rest_framework/pagination.py | |
| parent | 6fa589fefd48d98e4f0a11548b6c3e5ced58e31e (diff) | |
| download | django-rest-framework-b16fb5777168246b1e217640b818a82eb6e2141b.tar.bz2 | |
Expand pagination support, add docs
Diffstat (limited to 'rest_framework/pagination.py')
| -rw-r--r-- | rest_framework/pagination.py | 62 | 
1 files changed, 54 insertions, 8 deletions
| diff --git a/rest_framework/pagination.py b/rest_framework/pagination.py index 398e6f3d..f8b1fd1a 100644 --- a/rest_framework/pagination.py +++ b/rest_framework/pagination.py @@ -4,27 +4,64 @@ from rest_framework import serializers  class NextPageField(serializers.Field): +    """ +    Field that returns a link to the next page in paginated results. +    """      def to_native(self, value):          if not value.has_next():              return None          page = value.next_page_number() -        request = self.context['request'] -        return request.build_absolute_uri('?page=%d' % page) +        request = self.context.get('request') +        relative_url = '?page=%d' % page +        if request: +            return request.build_absolute_uri(relative_url) +        return relative_url  class PreviousPageField(serializers.Field): +    """ +    Field that returns a link to the previous page in paginated results. +    """      def to_native(self, value):          if not value.has_previous():              return None          page = value.previous_page_number() -        request = self.context['request'] -        return request.build_absolute_uri('?page=%d' % page) +        request = self.context.get('request') +        relative_url = '?page=%d' % page +        if request: +            return request.build_absolute_uri('?page=%d' % page) +        return relative_url -class PaginationSerializer(serializers.Serializer): -    count = serializers.Field(source='paginator.count') -    next = NextPageField(source='*') -    previous = PreviousPageField(source='*') +class PaginationSerializerOptions(serializers.SerializerOptions): +    """ +    An object that stores the options that may be provided to a +    pagination serializer by using the inner `Meta` class. + +    Accessible on the instance as `serializer.opts`. +    """ +    def __init__(self, meta): +        super(PaginationSerializerOptions, self).__init__(meta) +        self.object_serializer_class = getattr(meta, 'object_serializer_class', +                                               serializers.Field) + + +class BasePaginationSerializer(serializers.Serializer): +    """ +    A base class for pagination serializers to inherit from, +    to make implementing custom serializers more easy. +    """ +    _options_class = PaginationSerializerOptions +    _results_field = 'results' + +    def __init__(self, *args, **kwargs): +        """ +        Override init to add in the object serializer field on-the-fly. +        """ +        super(BasePaginationSerializer, self).__init__(*args, **kwargs) +        results_field = self._results_field +        object_serializer = self.opts.object_serializer_class +        self.fields[results_field] = object_serializer(source='object_list')      def to_native(self, obj):          """ @@ -32,3 +69,12 @@ class PaginationSerializer(serializers.Serializer):          each in turn.          """          return self.convert_object(obj) + + +class PaginationSerializer(BasePaginationSerializer): +    """ +    A default implementation of a pagination serializer. +    """ +    count = serializers.Field(source='paginator.count') +    next = NextPageField(source='*') +    previous = PreviousPageField(source='*') | 
