原生实现
示例,展示了如何创建一个统一返回格式和状态码的中间件:首先,在你的项目或应用中创建一个新的中间件文件,例如response_middleware.py
。
在response_middleware.py
中编写一个中间件类,如下所示:
from django.http import JsonResponse
class UnifiedResponseMiddleware:
def __init__(self, get_response):
self.get_response = get_response
def __call__(self, request):
response = self.get_response(request)
# 如果响应已经是一个JsonResponse,则不进行处理
if isinstance(response, JsonResponse):
return response
# 定义统一的响应格式
unified_response = {
'code': response.status_code,
'message': response.reason_phrase,
'data': response.content.decode('utf-8')
}
# 将原始响应替换为统一的JsonResponse
new_response = JsonResponse(unified_response)
# 将原始响应的状态码和其他头信息复制到新的JsonResponse中
new_response.status_code = response.status_code
for header, value in response.items():
new_response[header] = value
return new_response
settings.py
文件中的MIDDLEWARE
列表中:MIDDLEWARE = [
# ...
'yourapp.middleware.response_middleware.UnifiedResponseMiddleware',
# ...
]
这个示例中的UnifiedResponseMiddleware
中间件将统一响应格式为:
{
"code": 状态码,
"message": "状态消息",
"data": "响应内容"
}
注意,如果原始响应已经是一个JsonResponse
,这个示例中的中间件将不会进行处理。你可以根据自己的需求修改这个逻辑。
在生产环境中,你可能还需要考虑对响应数据进行适当的格式化,以保证返回的数据结构一致。此外,你可能需要为不同类型的响应(如HTML页面、JSON数据、文件下载等)添加特定的处理逻辑。
Django Rest Framework
框架,要实现统一的返回格式和状态码,你可以使用自定义的异常处理函数(Exception Handler)和响应渲染器(Response Renderer)来实现。utils
目录下),创建一个文件,例如custom_exception_handler.py
,并编写自定义异常处理函数:from rest_framework.views import exception_handler
from rest_framework.response import Response
def custom_exception_handler(exc, context):
response = exception_handler(exc, context)
if response is not None:
# 定义统一的响应格式
response.data = {
'code': response.status_code,
'message': response.status_text,
'data': response.data
}
return response
settings.py
文件中的REST_FRAMEWORK
设置中,将默认的异常处理函数替换为自定义的异常处理函数:REST_FRAMEWORK = {
# ...
'EXCEPTION_HANDLER': 'yourapp.utils.custom_exception_handler.custom_exception_handler',
# ...
}
utils
目录下),创建一个文件,例如custom_renderer.py
,并编写自定义的响应渲染器:from rest_framework.renderers import JSONRenderer
class CustomJSONRenderer(JSONRenderer):
def render(self, data, accepted_media_type=None, renderer_context=None):
# 定义统一的响应格式
unified_response = {
'code': renderer_context['response'].status_code,
'message': renderer_context['response'].status_text,
'data': data
}
return super().render(unified_response, accepted_media_type, renderer_context)
settings.py
文件中的REST_FRAMEWORK
设置中,将默认的响应渲染器替换为自定义的响应渲染器:REST_FRAMEWORK = {
# ...
'DEFAULT_RENDERER_CLASSES': [
'yourapp.utils.custom_renderer.CustomJSONRenderer',
# ...
],
# ...
}
现在,Django Rest Framework将在处理API请求时使用自定义的异常处理函数和响应渲染器,从而实现统一的返回格式和状态码。这个示例展示了如何实现一个简单的自定义异常处理函数和响应渲染器。你可以根据自己的需求进行调整和优化。请注意,这个示例仅用于演示目的,在生产环境中需要对安全性和性能进行更严格的考虑。