小工具类

二次封装rest_framework.response的Response类

from rest_framework.response import Response

class APIResponse(Response):
    def __init__(self,data_status,data_msg,results=None,
                 status=None,headers=None,content_type=None,**kwargs):
        data = {
            'status':data_status,
            'msg':data_msg
        }
        if results is not None:
            data['results'] = results
        data.update(**kwargs)
        super().__init__(data=data,status=status,headers=headers,content_type=content_type)