drf 自定义返回数据格式

参考:https://www.jianshu.com/p/c0be24752584

参考原作者的写法,加了API的展示

修改permissions.py

复制代码
from rest_framework import permissions
from rest_framework.renderers import JSONRenderer


# 重新封装Response对象
def response_wrapper(code, msg, data):
    # 自定义一个返回字典
    return {'code': code, 'msg': msg, 'data': data}


class MyJSONRenderer(JSONRenderer):
    # 重构render方法
    def render(self, data, accepted_media_type=None, renderer_context=None):
        code = renderer_context['response'].status_code
        if renderer_context and code == 200:
            ret = response_wrapper(code=code, msg='success', data=data)

        else:
            ret = response_wrapper(code=code, msg='fail', data=data)
        # 返回JSON数据
        return super().render(ret, accepted_media_type, renderer_context)
复制代码

修改settings.py

REST_FRAMEWORK = {
    ...
    # drf提供的渲染类
    'DEFAULT_RENDERER_CLASSES': [
        'script1.permissions.MyJSONRenderer',
        'rest_framework.renderers.BrowsableAPIRenderer',
    ],
}

 

posted @   不明真相的吃瓜群众22  阅读(128)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 分享4款.NET开源、免费、实用的商城系统
· 全程不用写代码,我用AI程序员写了一个飞机大战
· MongoDB 8.0这个新功能碉堡了,比商业数据库还牛
· 白话解读 Dapr 1.15:你的「微服务管家」又秀新绝活了
· 上周热点回顾(2.24-3.2)
点击右上角即可分享
微信分享提示