Добавление корневого элемента в json-ответ (django-rest-framework)

Я пытаюсь определить лучший способ добавить корневой элемент во все ответы json, используя django и django-rest-framework.

Я думаю, что добавление настраиваемого рендеринга - лучший способ выполнить то, что я хочу достичь, и это то, к чему я придумал:

from rest_framework.renderers import JSONRenderer

class CustomJSONRenderer(JSONRenderer):
#override the render method
def render(self, data, accepted_media_type=None, renderer_context=None):
    #call super, as we really just want to mess with the data returned
    json_str = super(CustomJSONRenderer, self).render(data, accepted_media_type, renderer_context)
    root_element = 'contact'

    #wrap the json string in the desired root element
    ret = '{%s: %s}' % (root_element, json_str) 

    return ret

Теперь сложная часть динамически устанавливает root_element на основе представления, из которого вызывается render().

Приветствуются любые указатели/советы,

Приветствия

Ответ 1

Для потомства ниже - окончательное решение. Он немного вырос от оригинала, так как теперь он переформатирует и разбитые на страницы результаты.

Также я должен был указать ранее, что причина для корневого элемента JSON заключается в интеграции с конечным решением Ember.

сериализатору:

from rest_framework.serializers import ModelSerializer
from api.models import Contact

class ContactSerializer(ModelSerializer):
    class Meta:
        model = Contact
        #define the resource we wish to use for the root element of the response
        resource_name = 'contact' 
        fields = ('id', 'first_name', 'last_name', 'phone_number', 'company')

визуализатор:

from rest_framework.renderers import JSONRenderer

class CustomJSONRenderer(JSONRenderer):
"""
    Override the render method of the django rest framework JSONRenderer to allow the following:
    * adding a resource_name root element to all GET requests formatted with JSON
    * reformatting paginated results to the following structure {meta: {}, resource_name: [{},{}]}

    NB: This solution requires a custom pagination serializer and an attribute of 'resource_name'
        defined in the serializer
"""
def render(self, data, accepted_media_type=None, renderer_context=None):
    response_data = {}

    #determine the resource name for this request - default to objects if not defined
    resource = getattr(renderer_context.get('view').get_serializer().Meta, 'resource_name', 'objects')

    #check if the results have been paginated
    if data.get('paginated_results'):
        #add the resource key and copy the results
        response_data['meta'] = data.get('meta')
        response_data[resource] = data.get('paginated_results')
    else:
        response_data[resource] = data

    #call super to render the response
    response = super(CustomJSONRenderer, self).render(response_data, accepted_media_type, renderer_context)

    return response

нумерацией страниц:

from rest_framework import pagination, serializers

class CustomMetaSerializer(serializers.Serializer):
    next_page = pagination.NextPageField(source='*')
    prev_page = pagination.PreviousPageField(source='*')
    record_count = serializers.Field(source='paginator.count')

class CustomPaginationSerializer(pagination.BasePaginationSerializer):
    # Takes the page object as the source
    meta = CustomMetaSerializer(source='*')
    results_field = 'paginated_results'

Ответ 2

Благодарим вас за то, что вы получили 95% от этого.

Лично я хотел добавить метаданные к каждому запросу api для определенного объекта, независимо от того, была ли она разбита на страницы. Я также хотел просто передать объект dict, который я определил вручную.

Тонкий пользовательский рендеринг

class CustomJSONRenderer(renderers.JSONRenderer):
    def render(self, data, accepted_media_type=None, renderer_context=None):
        response_data = {}
        # Name the object list 
        object_list = 'results'
        try:
            meta_dict = getattr(renderer_context.get('view').get_serializer().Meta, 'meta_dict')
        except:
            meta_dict = dict()
        try:
            data.get('paginated_results')
            response_data['meta'] = data['meta']
            response_data[object_list] = data['results']
        except:
            response_data[object_list] = data
            response_data['meta'] = dict()
            # Add custom meta data
            response_data['meta'].update(meta_dict)
            # Call super to render the response
            response = super(CustomJSONRenderer, self).render(response_data, accepted_media_type, renderer_context)
        return response

Родительский сериализатор и пример просмотра

class MovieListSerializer(serializers.ModelSerializer):
    class Meta:
        model = Movie
        meta_dict = dict()
        meta_dict['foo'] = 'bar'

class MovieViewSet(generics.ListAPIView):
    queryset = Movie.objects.exclude(image__exact = "")
    serializer_class = MovieListSerializer
    permission_classes = (IsAdminOrReadOnly,)
    renderer_classes = (CustomJSONRenderer,)
    pagination_serializer_class = CustomPaginationSerializer
    paginate_by = 10