我試圖實現(xiàn)這個答案Response來為我的所有類創(chuàng)建一個自定義基ModelViewSet類。我的問題在于retrieve功能。如果Id我正在尋找的實例沒有找到，我似乎無法Response用我修改的response_format. 它仍然給出默認(rèn)響應(yīng)。我應(yīng)該把我的if病情改成什么？views.py：class ResponseInfo(object): def __init__(self, **args): self.response = { "message": args.get('message', 'success'), "error": args.get('error', ), "data": args.get('data', []), }class LanguageView(viewsets.ModelViewSet): def __init__(self, **kwargs): self.response_format = ResponseInfo().response super(LanguageView, self).__init__(**kwargs) permission_classes = [permissions.DjangoModelPermissions] queryset = Language.objects.all() serializer_class = LanguageSerializer def list(self, request, *args, **kwargs): # call the original 'list' to get the original response. response_data = super(LanguageView, self).list(request, *args, **kwargs) # customize the response data. self.response_format['data'] = response_data.data if not response_data.data: self.response_format['message'] = 'List is empty.' self.response_format['error'] = response_data.status_code return Response(self.response_format) def retrieve(self, request, *args, **kwargs): response_data = super(LanguageView, self).retrieve(request, *args, **kwargs) self.response_format['data'] = response_data.data if not response_data.data: self.response_format['message'] = 'Instance not found.' self.response_format['error'] = response_data.status_code return Response(self.response_format)找到實例時的 JSON 響應(yīng)。例如http://127.0.0.1:8000/languages/1/：{ "message": "success", "error": null, "data": { "id": 1, "name": "English", "icon": "http://127.0.0.1:8000/media/language_icons/English.png", "xml": "http://127.0.0.1:8000/media/-", "abbreviation": "En" }}"error"沒有錯誤時是否可以不顯示變量？對于這種情況，當(dāng)列表不為空并且找到搜索的實例時。
1 回答
莫回?zé)o
TA貢獻(xiàn)1865條經(jīng)驗 獲得超7個贊
您可以實現(xiàn)自定義異常處理程序：
from rest_framework.views import exception_handler
def custom_exception_handler(exc, context):
# Call REST framework's default exception handler first,
# to get the standard error response.
response = exception_handler(exc, context)
# Now add the HTTP status code to the response.
if response is not None and response.status_code == 404:
response.data = {
"message": "Instance not found.",
"error": "HTTP_404_NOT_FOUND",
}
return response
要為您的項目應(yīng)用此處理程序，請將其添加到REST_FRAMEWORK設(shè)置中：
REST_FRAMEWORK = {
'EXCEPTION_HANDLER': 'my_project.my_app.utils.custom_exception_handler'
}
添加回答
舉報
0/150
提交
取消