python - 自定义 NotFound 异常 --- Django REST Framework

Question

假设一个最小的应用程序具有 person 模型、ModelSerializer 和相应的 ReadOnlyModelViewSet。

数据库中只存在一个条目，因此当请求 /person/1 时，正确的响应是:

{
  "name": "RandomName1"
}

当请求/person/2时，响应是:

{
  "detail": "Not found."
}

我想自定义这个。

尽管我读了 documentation目前尚不清楚我如何自定义它。

我应该澄清一下，我希望根据 View 对此进行自定义。例如 /person/2 应返回:

{
  "detail": "Person 2 was not found."
}

和/address/3应该返回:

{
  "detail": "Address 3 was not found."
}

Answer 1

听起来您想要处理具有 404 状态代码的响应。修改链接文档中的示例:

from rest_framework.views import exception_handler

def custom_exception_handler(exc, context):
    response = exception_handler(exc, context)

    if response.data['status_code'] == 404:
        try:
            response.data['detail'] = "{name} {id} was not found.".format(
                name=context['view'].verbose_name,
                id=context['kwargs']['id']  # this may need tweaking
            )
        except AttributeError:
            pass

    return response

然后将相应的verbose_name 添加到您的 View 中。