django中间件处理返回数据

统一接口返回数据格式的时候，每个视图都需要返回JsonResponse对象，比较麻烦，因此增加中间件处理视图返回的结果。

from django.http.response import JsonResponse, HttpResponseBase

code_msg_map = {
    0: 'ok',
    1: 'The args missing.',
    1000: 'The file does not exist.',
}


class ResponseMiddleware:
    """包装视图返回的数据，创建 JsonResponse 对象"""
    def __init__(self, get_response):
        self.get_response = get_response

    def __call__(self, request):
        response = self.get_response(request)
        # 包装返回的数据
        if not isinstance(response, HttpResponseBase):
            if isinstance(response, int):
                # 只返回错误码，没有数据，默认为 {}
                response = JsonResponse({'code': response, 'msg': code_msg_map.get(response, ''), 'data': {}})
            elif isinstance(response, tuple):
                # 包括错误码和数据
                code, data = response
                response = JsonResponse({'code': code, 'msg': code_msg_map.get(code, ''), 'data': data})
            else:
                # 只有数据，错误码默认为 0
                response = JsonResponse({'code': 0, 'msg': code_msg_map.get(0, ''), 'data': response})
        return response