Django+DRF 实战:自定义异常处理流程

一、DRF 异常处理流程

DRF 默认异常处理流程

DRF默认的异常处理流程如下:

  1. 当异常发生时,会自动调用rest_framework.views.exception_handler 函数来处理异常。
  2. 该函数对DRF异常(即继承了APIException 类)和Django内置的Http404、PermissionDenied异常,提取错误信息并返回响应
  3. 其它异常不处理,返回None(抛出到外层)。

源码

下面是DRF的exception_handler 异常处理函数源码

python 复制代码
# rest_framework.views.py
def exception_handler(exc, context):
    if isinstance(exc, Http404):
        exc = exceptions.NotFound(*(exc.args))
    elif isinstance(exc, PermissionDenied):
        exc = exceptions.PermissionDenied(*(exc.args))

    if isinstance(exc, exceptions.APIException):
        headers = {}
        if getattr(exc, 'auth_header', None):
            headers['WWW-Authenticate'] = exc.auth_header
        if getattr(exc, 'wait', None):
            headers['Retry-After'] = '%d' % exc.wait

        if isinstance(exc.detail, (list, dict)):
            data = exc.detail
        else:
            data = {'detail': exc.detail}

        set_rollback()
        return Response(data, status=exc.status_code, headers=headers)

    return None

二、实战

DRF 自定义异常处理流程

DRF 自定义异常处理流程示例:

  1. 定义custom_exception_handler函数,作为DRF的全局异常处理器
  2. 该函数记录详细的错误日志,包括用户、IP、请求路径、方法、视图及异常信息。然后,统一返回错误响应格式
    • 若异常未被 DRF 捕获,返回 500 错误或特定 ProtectedError 提示
    • 若为 DRF 的 ValidationError 或 Django 的验证错误,调用对应处理函数提取错误信息
    • 其他异常则根据状态码返回预定义的友好提示

应用自定义异常处理流程

settings.py 中,配置DRF自定义异常处理函数,作为全局异常处理器。

python 复制代码
REST_FRAMEWORK = {
    # ...
    # 全局异常处理
    "EXCEPTION_HANDLER": "mars_framework.exceptions.base.custom_exception_handler",
    # ...
}

点击查看完成代码


您正在阅读的是《Django从入门到实战》专栏!关注不迷路~

相关推荐
似水流年流不尽思念5 分钟前
Spring Bean有哪些生命周期回调方法?有哪几种实现方式?
后端·spring·面试
Moonbit12 分钟前
提交即有奖!MGPIC 游戏赛道官方推荐框架上线,直播同步解读赛题。 MoonBit MoonBit
后端·微信·程序员
郝同学的测开笔记13 分钟前
打通回家之路:OpenVPN,你的企业网络万能钥匙(一)
运维·后端·测试
whitepure13 分钟前
万字详解Java代码块
java·后端
SimonKing43 分钟前
Spring Boot Admin:一站式监控微服务,这个运维神器真香!
java·后端·程序员
uhakadotcom1 小时前
如何安装和使用开源的Meilisearch
后端·面试·github
高松燈1 小时前
自动拆箱 导致的空指针问题复盘
后端
IT_陈寒1 小时前
Java性能优化实战:5个立竿见影的技巧让你的应用提速50%
前端·人工智能·后端
陈随易2 小时前
10年老前端,分享20+严选技术栈
前端·后端·程序员
汪子熙3 小时前
计算机世界里的 blob:从数据库 BLOB 到 Git、Web API 与云存储的二进制宇宙
后端