Django重写JSONRenderer类自定义返回的Json数据格式

重写JSONRenderer类，类似于django的中间件，在response返回中viewset和试图就会穿过render方法，在返回字段里增加了code和msg参数

from rest_framework.renderers import JSONRenderer
class CustomJsonRender(JSONRenderer):
    """ 自定义返回数据 Json格式 { "code": 0, "msg": "success", "data": { ... } } """

    def render(self, data, accepted_media_type=None, renderer_context=None):
        if renderer_context:
            response = renderer_context['response']
            code = 0 if int(response.status_code / 100) == 2 else response.status_code
            msg = 'success'
            if isinstance(data, dict):
                msg = data.pop('msg', msg)
                code = data.pop('code', code)
                data = data.pop('data', data)
            if code != 0 and data:
                msg = data.pop('detail', 'failed')
            response.status_code = 200
            res = {
                'code': code,
                'msg': msg,
                'data': data,
            }
            print(res)
            return super().render(res, accepted_media_type, renderer_context)
        else:
            return super().render(data, accepted_media_type, renderer_context)

• 局部使用

class RouteView(ModelViewSet):
    queryset = models.UserInfo.objects.all()
    serializer_class = RouteSerializer
    renderer_classes = (CustomJsonRender,)
    search_fields = ['user']

• 全局使用：在setting中配置

REST_FRAMEWORK = {
    'UNAUTHENTICATED_USER': None,
    'UNAUTHENTICATED_TOKEN': None,
    "DEFAULT_THROTTLE_RATES": {
        'luffy_anon': '5/m',
        'luffy_user': '10/m'
    },
    'DEFAULT_FILTER_BACKENDS': ('rest_framework.filters.DjangoFilterBackend','rest_framework.filters.SearchFilter'),
    'DEFAULT_RENDERER_CLASSES': ('rest_framework.renderers.JSONRenderer',)
}

结束！