From a0f40d99708790a422c7aa6fea0676df12b1db5a Mon Sep 17 00:00:00 2001
From: Simon <simobilleter@gmail.com>
Date: Sun, 6 Jul 2025 16:36:16 +0700
Subject: [PATCH] backend bulk delete filter

---
 backend/download/serializers.py |  4 ++++
 backend/download/src/queue.py   | 12 ++++++++++--
 backend/download/views.py       | 11 ++++++++++-
 3 files changed, 24 insertions(+), 3 deletions(-)

diff --git a/backend/download/serializers.py b/backend/download/serializers.py
index 82ceed0e..5ab3eca8 100644
--- a/backend/download/serializers.py
+++ b/backend/download/serializers.py
@@ -53,6 +53,10 @@ class DownloadListQueueDeleteQuerySerializer(serializers.Serializer):
     """serialize bulk delete download queue query string"""
 
     filter = serializers.ChoiceField(choices=["pending", "ignore"])
+    channel = serializers.CharField(required=False, help_text="channel ID")
+    vid_type = serializers.ChoiceField(
+        choices=VideoTypeEnum.values_known(), required=False
+    )
 
 
 class AddDownloadItemSerializer(serializers.Serializer):
diff --git a/backend/download/src/queue.py b/backend/download/src/queue.py
index 7110ceaf..eaae66c3 100644
--- a/backend/download/src/queue.py
+++ b/backend/download/src/queue.py
@@ -100,9 +100,17 @@ class PendingInteract:
         path = f"ta_download/_doc/{self.youtube_id}"
         _, _ = ElasticWrap(path).delete(refresh=True)
 
-    def delete_by_status(self):
+    def delete_bulk(self, channel_id: str | None, vid_type: str | None):
         """delete all matching item by status"""
-        data = {"query": {"term": {"status": {"value": self.status}}}}
+        must_list = [{"term": {"status": {"value": self.status}}}]
+        if channel_id:
+            must_list.append({"term": {"channel_id": {"value": channel_id}}})
+
+        if vid_type:
+            must_list.append({"term": {"vid_type": {"value": vid_type}}})
+
+        data = {"query": {"bool": {"must": must_list}}}
+
         path = "ta_download/_delete_by_query"
         _, _ = ElasticWrap(path).post(data=data)
 
diff --git a/backend/download/views.py b/backend/download/views.py
index 963b6365..b3cec1a9 100644
--- a/backend/download/views.py
+++ b/backend/download/views.py
@@ -138,9 +138,18 @@ class DownloadApiListView(ApiBaseView):
         validated_query = serializer.validated_data
 
         query_filter = validated_query["filter"]
+        channel = validated_query.get("channel")
+        vid_type = validated_query.get("vid_type")
         message = f"delete queue by status: {query_filter}"
+        if channel:
+            message += f" - filter by channel: {channel}"
+        if vid_type:
+            message += f" - filter by vid_type: {vid_type}"
+
         print(message)
-        PendingInteract(status=query_filter).delete_by_status()
+        PendingInteract(status=query_filter).delete_bulk(
+            channel_id=channel, vid_type=vid_type
+        )
 
         return Response(status=204)