From f7b508298304d41bc8ff52fdec3ba0f2346d2e31 Mon Sep 17 00:00:00 2001
From: simon <simobilleter@gmail.com>
Date: Fri, 3 Dec 2021 21:05:34 +0700
Subject: [PATCH] use message:playlistscan for index_channel_playlists

---
 tubearchivist/home/tasks.py | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/tubearchivist/home/tasks.py b/tubearchivist/home/tasks.py
index 8a7c505..651eb06 100644
--- a/tubearchivist/home/tasks.py
+++ b/tubearchivist/home/tasks.py
@@ -44,7 +44,7 @@ def update_subscribed():
         "status": "message:rescan",
         "level": "info",
         "title": "Rescanning channels and playlists.",
-        "message": "Looking for new videos."
+        "message": "Looking for new videos.",
     }
     RedisArchivist().set_message("message:rescan", message)
 
@@ -259,6 +259,15 @@ def subscribe_to(url_str):
 def index_channel_playlists(channel_id):
     """add all playlists of channel to index"""
     channel_handler = YoutubeChannel(channel_id)
+    channel_name = channel_handler.channel_dict["channel_name"]
+    # notify
+    mess_dict = {
+        "status": "message:playlistscan",
+        "level": "info",
+        "title": "Looking for playlists",
+        "message": f'Scanning channel "{channel_name}" in progress',
+    }
+    RedisArchivist().set_message("message:playlistscan", mess_dict)
     all_playlists = channel_handler.get_all_playlists()
 
     if not all_playlists:
@@ -268,7 +277,16 @@ def index_channel_playlists(channel_id):
     all_indexed = PendingList().get_all_indexed()
     all_youtube_ids = [i["youtube_id"] for i in all_indexed]
 
+    counter = 1
     for playlist_id, playlist_title in all_playlists:
+        # notify
+        mess_dict = {
+            "status": "message:playlistscan",
+            "level": "info",
+            "title": "Scanning channel for playlists",
+            "message": f"Progress: {counter}/{len(all_playlists)}",
+        }
+        RedisArchivist().set_message("message:playlistscan", mess_dict)
         print("add playlist: " + playlist_title)
         playlist_handler = YoutubePlaylist(
             playlist_id, all_youtube_ids=all_youtube_ids
@@ -287,6 +305,7 @@ def index_channel_playlists(channel_id):
             continue
         playlist_handler.upload_to_es()
         playlist_handler.add_vids_to_playlist()
+        counter = counter + 1
 
     if all_playlists:
         handler = ThumbManager()