封装Response对象

封装Response对象

#utils.py
#自定制响应
from rest_framework.response import Response

class CommonResponse(Response):
    def __init__(self, code=10, msg='成功', data=None, **kwargs):
        dic = {'code': code, 'msg': msg, 'data': data}
        if data:
            dic = {'code': code, 'msg': msg, 'data': data}
        dic.update(kwargs)
        super().__init__(data=dic, status=None, headers=None, )

#urls.py
path('test7/', views.Test7View.as_view()),

#views.py
from app01.utils import CommonResponse
class Test7View(APIView):
    def get(self, request, *args, **kwargs):
        return CommonResponse(data={'name':'zhao'},demo='kfdha',token='fldksjafld')

#views.py
from app01.utils import CommonResponse
class Test7View(APIView):
    def get(self, request, *args, **kwargs):
       
        return CommonResponse(token='fldksjafld')

class Test7View(APIView):
    def get(self, request, *args, **kwargs):
   
        return CommonResponse(code=101,msg='错误',data={'name':'lisi'},aa='falkda')