django rest framework - custom renderer



我正在尝试调整现有的XMLRenderer以创建自定义的 -

class CustomRenderer(renderers.BaseRenderer):
    """
    Renderer which serializes to CustomXML.
    """
    media_type = 'application/xml'
    format = 'xml'
    charset = 'utf-8'
    def render(self, data, accepted_media_type=None, renderer_context=None):
        """
        Renders *obj* into serialized XML.
        """
        if data is None:
            return ''
        stream = StringIO()
        xml = SimplerXMLGenerator(stream, self.charset)
        xml.startDocument()
        xml.startElement("job id='string1'", {})
        self._to_xml(xml, data)
        xml.endElement("job")
        xml.endDocument()
        return stream.getvalue()
    def _to_xml(self, xml, data):
        if isinstance(data, (list, tuple)):
            for item in data:
                xml.startElement("string2", {})
                self._to_xml(xml, item)
                xml.endElement("string2")
        elif isinstance(data, dict):
            for key, value in six.iteritems(data):
                xml.startElement(key, {})
                self._to_xml(xml, value)
                xml.endElement(key)
        elif data is None:
            # Don't output any value
            pass
        else:
            xml.characters(smart_text(data))

对于 string1,我希望它从调用它的视图中获取值。 string1 = API 中 GET 的主键。

即,如果我打电话给http://localhost/API/2345 then string1 = 2345

对于 string2,我希望它返回类似于他们在以下帖子中所做的模型名称 -

将根元素添加到 json 响应(django-rest-framework)

这是自定义渲染器返回的值,以便可以将 JSON/XML 的根值设置为模型名称。

我尝试调整自定义渲染器以包含行,但是在运行视图时,它抱怨我的 views.py"object() 不带任何参数" -

if request.method == 'GET':
    DEV = Trgjob.objects.using('database1').filter(job_id=pk).order_by('job_order')
    serializer = CustomSerializer(DEV, many=True)
    return CustomRenderer(serializer.data)

看起来您正在尝试从视图中返回渲染器实例?您应该只从视图中返回常规数据,但使用视图类上的 renderer_classes 属性设置渲染器类。

class MyView(APIView):
    renderer_classes = [CustomRenderer]
    ...
    def get(self, request, pk=None):
        DEV = Trgjob.objects.using('database1').filter(job_id=pk).order_by('job_order')
        serializer = CustomSerializer(DEV, many=True)
        return Response(serializer.data)

编辑:至于从视图中将额外的上下文放入渲染器中...您可以通过检查renderer_context['view']来做到这一点,因此在您的render()方法中具有以下含义...

view = renderer_context['view']
job_id = view.kwargs['job_id']

最新更新