修改Djangorestframework响应结构

Question

我遇到了这样的情况:我的前卫需要我的所有端点以下面的格式做出响应。

{ status: 200, message: "OK", content: {Normal DRF Response Body} }

我知道我可以通过在我的所有视图中使用APIView来获得这个结构，但是这样我就不能使用泛型视图或视图集了。

我在想，是否有任何方法来扩展DRF响应类，并在settings.py中指定扩展类，或者为此目的使用中间件。

我们能做些什么呢？

Answer 1

是的，有。

您可以实现呈现器并将其添加到settings.py中，如下所示：

REST_FRAMEWORK = {
    'DEFAULT_RENDERER_CLASSES': (
        '< your renderer >',       
    ),  
}

对于实际的呈现器--从rest_framework的JSON呈现器( rest_framework.renderers.JSONRenderer )中获取灵感。

您基本上可以接受这个JSONRenderer的实现，并稍微修改呈现函数。

  def render(self, data, accepted_media_type=None, renderer_context=None):
    """
    Render `data` into JSON, returning a bytestring.
    """

在这里，您可以访问数据和renderer_context。上下文有响应对象，这是您的响应。它具有'status_code‘属性，您可以在响应中使用它。

response = renderer_context['response']
my_response = { "status" : response.status_code,
                "message" : "OK",
                "content" : data }

对于该消息，取决于您在何处获得有关给定状态代码的正确消息。

然后，如果您遵循原始的JSONRendener实现，您可以找到以下片段，您只需将其更改为转储您的新响应：

ret = json.dumps(
        my_response, cls=self.encoder_class,
        indent=indent, ensure_ascii=self.ensure_ascii,
        allow_nan=not self.strict, separators=separators
    )