From eb6d6be3b96360cb554120c7c97997e02b4e33c2 Mon Sep 17 00:00:00 2001 From: simon Date: Sun, 17 Apr 2022 10:08:24 +0700 Subject: [PATCH] handle 404 in channel-video and playlist-video api view --- tubearchivist/api/views.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/tubearchivist/api/views.py b/tubearchivist/api/views.py index 72dfeaa..7cd7a2a 100644 --- a/tubearchivist/api/views.py +++ b/tubearchivist/api/views.py @@ -52,7 +52,10 @@ class ApiBaseView(APIView): print(self.search_base) response, status_code = ElasticWrap(self.search_base).get(data=data) self.response["data"] = SearchProcess(response).process() - self.status_code = status_code + if self.response["data"]: + self.status_code = status_code + else: + self.status_code = 404 class VideoApiView(ApiBaseView): @@ -238,7 +241,7 @@ class ChannelApiVideoView(ApiBaseView): self.get_document_list(data) self.get_paginate() - return Response(self.response) + return Response(self.response, status=self.status_code) class PlaylistApiListView(ApiBaseView): @@ -286,7 +289,7 @@ class PlaylistApiVideoView(ApiBaseView): } self.get_document_list(data) self.get_paginate() - return Response(self.response) + return Response(self.response, status=self.status_code) class DownloadApiView(ApiBaseView):