public/tubearchivist
1
0
Fork 0
mirror of https://git.vectorsigma.ru/public/tubearchivist.git synced 2026-08-04 21:39:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

backend bulk delete filter

Browse Source
This commit is contained in:
Simon
2025-07-06 16:36:16 +07:00
parent 8d9cb9261e
commit a0f40d9970
3 changed files with 24 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
backend/download/serializers.py
View File

@@ -53,6 +53,10 @@ class DownloadListQueueDeleteQuerySerializer(serializers.Serializer):
"""serialize bulk delete download queue query string""" """serialize bulk delete download queue query string"""
filter = serializers.ChoiceField(choices=["pending", "ignore"]) 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): class AddDownloadItemSerializer(serializers.Serializer):

12
backend/download/src/queue.py
View File

@@ -100,9 +100,17 @@ class PendingInteract:
path = f"ta_download/_doc/{self.youtube_id}" path = f"ta_download/_doc/{self.youtube_id}"
_, _ = ElasticWrap(path).delete(refresh=True) _, _ = 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""" """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" path = "ta_download/_delete_by_query"
_, _ = ElasticWrap(path).post(data=data) _, _ = ElasticWrap(path).post(data=data)

11
backend/download/views.py
View File

@@ -138,9 +138,18 @@ class DownloadApiListView(ApiBaseView):
validated_query = serializer.validated_data validated_query = serializer.validated_data
query_filter = validated_query["filter"] 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}" 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) 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) return Response(status=204)