[Python知识库]drf---Response重写

# response_expestion.py

from rest_framework.views import exception_handler
from rest_framework import status
from rest_framework.response import Response


class APIResponse(Response):
    def __init__(self, result=None, code=None, msg='成功', status=None, headers=None, content_type=None, **kwargs):
        dic = {
            'code': code,
            'msg': msg
        }
        if result:
            dic['result'] = result
        else:
            dic.update(kwargs)
        super().__init__(data=dict, status=status, headers=headers, content_type=None)