better playlist scan message, private methods in YoutubeChannel class

This commit is contained in:
simon 2022-03-22 10:07:33 +07:00
parent 3d451e47e5
commit 1f1dfcb54f
No known key found for this signature in database
GPG Key ID: 2C15AA5E89985DD4
2 changed files with 10 additions and 7 deletions

View File

@ -247,11 +247,13 @@ class YoutubeChannel(YouTubeItem):
def index_channel_playlists(self): def index_channel_playlists(self):
"""add all playlists of channel to index""" """add all playlists of channel to index"""
print(f"{self.youtube_id}: index all playlists") print(f"{self.youtube_id}: index all playlists")
self.get_from_es()
channel_name = self.json_data["channel_name"]
mess_dict = { mess_dict = {
"status": "message:playlistscan", "status": "message:playlistscan",
"level": "info", "level": "info",
"title": "Looking for playlists", "title": "Looking for playlists",
"message": f'Scanning channel "{self.youtube_id}" in progress', "message": f"{channel_name}: Scanning channel in progress",
} }
RedisArchivist().set_message("message:playlistscan", mess_dict) RedisArchivist().set_message("message:playlistscan", mess_dict)
self.get_all_playlists() self.get_all_playlists()
@ -261,22 +263,23 @@ class YoutubeChannel(YouTubeItem):
all_youtube_ids = self.get_all_video_ids() all_youtube_ids = self.get_all_video_ids()
for idx, playlist in enumerate(self.all_playlists): for idx, playlist in enumerate(self.all_playlists):
self.notify_single_playlist(idx, playlist) self._notify_single_playlist(idx, playlist)
self.index_single_playlist(playlist, all_youtube_ids) self._index_single_playlist(playlist, all_youtube_ids)
def notify_single_playlist(self, idx, playlist): def _notify_single_playlist(self, idx, playlist):
"""send notification""" """send notification"""
channel_name = self.json_data["channel_name"]
mess_dict = { mess_dict = {
"status": "message:playlistscan", "status": "message:playlistscan",
"level": "info", "level": "info",
"title": "Scanning channel for playlists", "title": f"{channel_name}: Scanning channel for playlists",
"message": f"Progress: {idx + 1}/{len(self.all_playlists)}", "message": f"Progress: {idx + 1}/{len(self.all_playlists)}",
} }
RedisArchivist().set_message("message:playlistscan", mess_dict) RedisArchivist().set_message("message:playlistscan", mess_dict)
print("add playlist: " + playlist[1]) print("add playlist: " + playlist[1])
@staticmethod @staticmethod
def index_single_playlist(playlist, all_youtube_ids): def _index_single_playlist(playlist, all_youtube_ids):
"""add single playlist if needed""" """add single playlist if needed"""
playlist = YoutubePlaylist(playlist[0]) playlist = YoutubePlaylist(playlist[0])
playlist.all_youtube_ids = all_youtube_ids playlist.all_youtube_ids = all_youtube_ids

View File

@ -7,7 +7,7 @@ checkMessages()
// page map to notification status // page map to notification status
const messageTypes = { const messageTypes = {
"download": ["message:download", "message:add", "message:rescan"], "download": ["message:download", "message:add", "message:rescan", "message:playlistscan"],
"channel": ["message:subchannel"], "channel": ["message:subchannel"],
"channel_id": ["message:playlistscan"], "channel_id": ["message:playlistscan"],
"playlist": ["message:subplaylist"], "playlist": ["message:subplaylist"],