diff options
Diffstat (limited to 'rest_framework/generics.py')
| -rw-r--r-- | rest_framework/generics.py | 55 |
1 files changed, 45 insertions, 10 deletions
diff --git a/rest_framework/generics.py b/rest_framework/generics.py index 4240e33e..8647ad42 100644 --- a/rest_framework/generics.py +++ b/rest_framework/generics.py @@ -2,7 +2,8 @@ Generic views that provide commmonly needed behaviour. """ -from rest_framework import views, mixins, serializers +from rest_framework import views, mixins +from rest_framework.settings import api_settings from django.views.generic.detail import SingleObjectMixin from django.views.generic.list import MultipleObjectMixin @@ -14,22 +15,39 @@ class BaseView(views.APIView): Base class for all other generic views. """ serializer_class = None + model_serializer_class = api_settings.MODEL_SERIALIZER - def get_serializer(self, data=None, files=None, instance=None): - # TODO: add support for files - # TODO: add support for seperate serializer/deserializer + def get_serializer_context(self): + """ + Extra context provided to the serializer class. + """ + return { + 'request': self.request, + 'format': self.format, + 'view': self + } + + def get_serializer_class(self): + """ + Return the class to use for the serializer. + Use `self.serializer_class`, falling back to constructing a + model serializer class from `self.model_serializer_class` + """ serializer_class = self.serializer_class if serializer_class is None: - class DefaultSerializer(serializers.ModelSerializer): + class DefaultSerializer(self.model_serializer_class): class Meta: model = self.model serializer_class = DefaultSerializer - context = { - 'request': self.request, - 'format': self.kwargs.get('format', None) - } + return serializer_class + + def get_serializer(self, data=None, files=None, instance=None): + # TODO: add support for files + # TODO: add support for seperate serializer/deserializer + serializer_class = self.get_serializer_class() + context = self.get_serializer_context() return serializer_class(data, instance=instance, context=context) @@ -37,7 +55,24 @@ class MultipleObjectBaseView(MultipleObjectMixin, BaseView): """ Base class for generic views onto a queryset. """ - pass + + pagination_serializer_class = api_settings.PAGINATION_SERIALIZER + paginate_by = api_settings.PAGINATE_BY + + def get_pagination_serializer_class(self): + """ + Return the class to use for the pagination serializer. + """ + class SerializerClass(self.pagination_serializer_class): + class Meta: + object_serializer_class = self.get_serializer_class() + + return SerializerClass + + def get_pagination_serializer(self, page=None): + pagination_serializer_class = self.get_pagination_serializer_class() + context = self.get_serializer_context() + return pagination_serializer_class(instance=page, context=context) class SingleObjectBaseView(SingleObjectMixin, BaseView): |
