tubearchivist/tubearchivist
1
0
Fork 0
mirror of https://github.com/tubearchivist/tubearchivist.git synced 2024-12-22 18:00:13 +00:00
Code Issues Projects Releases Wiki Activity

implement skip_on_empty in update_playlist

Browse Source
This commit is contained in:
Simon 2024-05-11 23:09:03 +02:00
parent 97b6d7d606
commit f874d402b1
No known key found for this signature in database
GPG Key ID: 2C15AA5E89985DD4
2 changed files with 10 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tubearchivist/home/src/index/channel.py
View File

@ -287,18 +287,7 @@ class YoutubeChannel(YouTubeItem):
def _index_single_playlist(playlist): def _index_single_playlist(playlist):
"""add single playlist if needed""" """add single playlist if needed"""
playlist = YoutubePlaylist(playlist[0]) playlist = YoutubePlaylist(playlist[0])
playlist.build_json() playlist.update_playlist(skip_on_empty=True)
if not playlist.json_data:
return
entries = playlist.json_data["playlist_entries"]
downloaded = [i for i in entries if i["downloaded"]]
if not downloaded:
return
playlist.upload_to_es()
playlist.add_vids_to_playlist()
playlist.get_playlist_art()
def get_channel_videos(self): def get_channel_videos(self):
"""get all videos from channel""" """get all videos from channel"""

10
tubearchivist/home/src/index/playlist.py
View File

@ -160,16 +160,24 @@ class YoutubePlaylist(YouTubeItem):
if status_code == 200: if status_code == 200:
print(f"{self.youtube_id}: removed {video_id} from playlist") print(f"{self.youtube_id}: removed {video_id} from playlist")
def update_playlist(self): def update_playlist(self, skip_on_empty=False):
"""update metadata for playlist with data from YouTube""" """update metadata for playlist with data from YouTube"""
self.build_json(scrape=True) self.build_json(scrape=True)
if not self.json_data: if not self.json_data:
# return false to deactivate # return false to deactivate
return False return False
if skip_on_empty:
has_item_downloaded = next(
i["downloaded"] for i in self.json_data["playlist_entries"]
)
if not has_item_downloaded:
return True
self.upload_to_es() self.upload_to_es()
self.add_vids_to_playlist() self.add_vids_to_playlist()
self.remove_vids_from_playlist() self.remove_vids_from_playlist()
self.get_playlist_art()
return True return True
def build_nav(self, youtube_id): def build_nav(self, youtube_id):