Django自定义封装Response

Django自定义封装Response

定义Response类

# utils/common_response.py
from rest_framework.response import Response

# 继承DRF的Response并重写__init__
class APIResponse(Response):
    def __init__(self, code=100, msg='成功', status=None, headers=None, **kwargs):
        data = {'code': code, 'msg': msg}
        if kwargs:
            data.update(kwargs)
        # 调用父类的构造方法并重新初始化参数
        super().__init__(data=data, status=status, headers=headers)

视图层示例

from rest_framework.views import APIView
from utils.common_response import APIResponse


class LoggerView(APIView):
    def get(self, request):
        info = ('这是一条info级别日志')
        return APIResponse(msg=info, headers={'name': 'Anna'})