From 40bb3e880ecc18a180144051f32f11db0cf00597 Mon Sep 17 00:00:00 2001 From: simon Date: Sun, 17 Apr 2022 19:15:40 +0700 Subject: [PATCH] API: implement status update and delete of item in queue --- tubearchivist/api/README.md | 19 ++++++++++++++++++- tubearchivist/api/views.py | 29 ++++++++++++++++++++++++++++- 2 files changed, 46 insertions(+), 2 deletions(-) diff --git a/tubearchivist/api/README.md b/tubearchivist/api/README.md index f6e10f4..2067331 100644 --- a/tubearchivist/api/README.md +++ b/tubearchivist/api/README.md @@ -149,8 +149,25 @@ POST /api/download/ ``` ## Download Queue Item View -/api/download/\/ +GET /api/download/\/ +POST /api/download/\/ +Ignore video in download queue: +```json +{ + "status": "ignore" +} +``` + +Add to queue previously ignored video: +```json +{ + "status": "pending" +} +``` + +DELETE /api/download/\/ +Forget or delete from download queue ## Ping View Validate your connection with the API diff --git a/tubearchivist/api/views.py b/tubearchivist/api/views.py index 7cd7a2a..a610c56 100644 --- a/tubearchivist/api/views.py +++ b/tubearchivist/api/views.py @@ -1,11 +1,12 @@ """all API views""" from api.src.search_processor import SearchProcess +from home.src.download.queue import PendingInteract from home.src.es.connect import ElasticWrap from home.src.index.video import SponsorBlock from home.src.ta.config import AppConfig from home.src.ta.helper import UrlListParser -from home.src.ta.ta_redis import RedisArchivist +from home.src.ta.ta_redis import RedisArchivist, RedisQueue from home.tasks import extrac_dl, subscribe_to from rest_framework.authentication import ( SessionAuthentication, @@ -295,9 +296,12 @@ class PlaylistApiVideoView(ApiBaseView): class DownloadApiView(ApiBaseView): """resolves to /api/download// GET: returns metadata dict of an item in the download queue + POST: update status of item to pending or ignore + DELETE: forget from download queue """ search_base = "ta_download/_doc/" + valid_status = ["pending", "ignore"] def get(self, request, video_id): # pylint: disable=unused-argument @@ -305,6 +309,29 @@ class DownloadApiView(ApiBaseView): self.get_document(video_id) return Response(self.response, status=self.status_code) + def post(self, request, video_id): + """post to video to change status""" + item_status = request.data["status"] + if item_status not in self.valid_status: + message = f"{video_id}: invalid status {item_status}" + print(message) + return Response({"message": message}, status=400) + + print(f"{video_id}: change status to {item_status}") + PendingInteract(video_id=video_id, status=item_status).update_status() + RedisQueue().clear_item(video_id) + + return Response(request.data) + + @staticmethod + def delete(request, video_id): + # pylint: disable=unused-argument + """delete single video from queue""" + print(f"{video_id}: delete from queue") + PendingInteract(video_id=video_id).delete_item() + + return Response({"success": True}) + class DownloadApiListView(ApiBaseView): """resolves to /api/download/