From ad8d75c3e8f6bd5b9ab6a91776e0c91854ec2c5d Mon Sep 17 00:00:00 2001
From: simon <simobilleter@gmail.com>
Date: Fri, 17 Dec 2021 23:28:53 +0700
Subject: [PATCH] add autodelete videos to ignore list, #112

---
 tubearchivist/home/src/download.py | 19 +++++++++++++------
 1 file changed, 13 insertions(+), 6 deletions(-)

diff --git a/tubearchivist/home/src/download.py b/tubearchivist/home/src/download.py
index 2fa6f4e1..454d3ef4 100644
--- a/tubearchivist/home/src/download.py
+++ b/tubearchivist/home/src/download.py
@@ -75,7 +75,7 @@ class PendingList:
 
         return missing_videos
 
-    def add_to_pending(self, missing_videos):
+    def add_to_pending(self, missing_videos, ignore=False):
         """build the bulk json data from pending"""
         # check if channel is indexed
         channel_handler = ChannelSubscription()
@@ -84,7 +84,7 @@ class PendingList:
         # check if already there
         self.all_downloaded = self.get_all_downloaded()
 
-        bulk_list, all_videos_added = self.build_bulk(missing_videos)
+        bulk_list, all_videos_added = self.build_bulk(missing_videos, ignore)
         # add last newline
         bulk_list.append("\n")
         query_str = "\n".join(bulk_list)
@@ -99,7 +99,7 @@ class PendingList:
 
         return all_videos_added
 
-    def build_bulk(self, missing_videos):
+    def build_bulk(self, missing_videos, ignore=False):
         """build the bulk lists"""
         bulk_list = []
         all_videos_added = []
@@ -116,7 +116,10 @@ class PendingList:
 
             channel_indexed = video["channel_id"] in self.all_channel_ids
             video["channel_indexed"] = channel_indexed
-            video["status"] = "pending"
+            if ignore:
+                video["status"] = "ignore"
+            else:
+                video["status"] = "pending"
             action = {"create": {"_id": youtube_id, "_index": "ta_download"}}
             bulk_list.append(json.dumps(action))
             bulk_list.append(json.dumps(video))
@@ -790,8 +793,12 @@ class VideoDownloader:
             "sort": [{"player.watched_date": {"order": "asc"}}],
         }
         all_to_delete = IndexPaginate("ta_video", data).get_results()
+        all_youtube_ids = [i["youtube_id"] for i in all_to_delete]
 
-        for to_delete in all_to_delete:
-            youtube_id = to_delete["youtube_id"]
+        for youtube_id in all_youtube_ids:
             print(f"autodelete {youtube_id}")
             YoutubeVideo(youtube_id).delete_media_file()
+
+        print("add deleted to ignore list")
+        pending_handler = PendingList()
+        pending_handler.add_to_pending(all_youtube_ids, ignore=True)