生产环境中处理Django Rest Framework 5XX错误，避免敏感信息泄露

DRF官方提供了可自定义的异常处理机制，这是比手动try-except或全局中间件更高效精准的方案，能完美实现你要的环境差异化响应需求。

核心逻辑 #

1. 替换DRF默认的exception_handler函数，统一接管所有未捕获的异常
2. 根据settings.DEBUG状态调整响应内容：
   • DEBUG=True：返回完整错误详情
   • DEBUG=False：仅返回500状态码，不携带响应体

具体实现 #

1. 编写自定义异常处理函数 #

在项目的工具模块（比如utils/exceptions.py）中添加：

from django.conf import settings
from rest_framework.views import exception_handler
from rest_framework.response import Response

def custom_exception_handler(exc, context):
    # 先调用DRF默认处理逻辑，获取基础响应对象
    response = exception_handler(exc, context)

    # 针对生产环境的5XX错误做处理
    if response is not None and response.status_code >= 500 and not settings.DEBUG:
        # 返回空响应体，仅保留状态码
        return Response(status=response.status_code)
    
    # DEBUG模式下保持原有错误详情响应
    return response

2. 配置DRF使用自定义处理函数 #

在settings.py的DRF配置中指定该函数：

REST_FRAMEWORK = {
    'EXCEPTION_HANDLER': 'utils.exceptions.custom_exception_handler',
}

3. 移除视图中的冗余try-except #

之前手动写的try-except可以全部删除，让DRF的异常处理机制自动捕获异常，既减少重复代码，又能统一控制响应格式。

针对手动抛出的5XX响应 #

如果业务场景需要手动返回5XX错误，可以封装一个工具函数统一处理：

from django.conf import settings
from rest_framework.response import Response
from rest_framework import status

def server_error_response(exception=None):
    if settings.DEBUG and exception:
        return Response(str(exception), status=status.HTTP_500_INTERNAL_SERVER_ERROR)
    return Response(status=status.HTTP_500_INTERNAL_SERVER_ERROR)

视图中调用示例：

try:
    # 业务逻辑代码
except Exception as e:
    return server_error_response(e)

方案优势 #

DRF的异常处理函数仅针对视图抛出的异常生效，不会遍历所有响应，性能损耗远低于全局中间件，完全符合官方推荐的最佳实践。

内容的提问来源于stack exchange，提问作者Leo