404错误的Django Rest Framework自定义消息

Question

我有一个基于类的通用视图：class ProjectDetails(mixins.RetrieveModelMixin,                     mixins.UpdateModelMixin,                     generics.GenericAPIView):    queryset = Project.objects.all()    # Rest of definition在我的urls.py，我有：urlpatterns = [    url(r'^(?P<pk>[0-9]+)/$', views.ProjectDetails.as_view())]使用不存在的ID调用API时，它将返回HTTP 404包含以下内容的响应：{    "detail": "Not found."}是否可以修改此响应？我只需要为此视图自定义错误消息。

Answer 1

通过覆盖诸如的特定方法是可能的update，retrieve例如：

from django.http import Http404

from rest_framework.response import Response

class ProjectDetails(mixins.RetrieveModelMixin,

                     mixins.UpdateModelMixin,

                     generics.GenericAPIView):

    queryset = Project.objects.all()

    def retrieve(self, request, *args, **kwargs):

        try:

            return super().retrieve(request, *args, **kwargs)

        except Http404:

            return Response(data={"cusom": "message"})

    def update(self, request, *args, **kwargs):

        try:

            return super().update(request, *args, **kwargs)

        except Http404:

            return Response(data={"cusom": "message"})