snegov/robocyp
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Print video title in add/add-csv methods

Browse Source
This commit is contained in:
Maks Snegov 2024-02-03 15:22:50 -08:00
parent 9faaea8186
commit 0be2379675
1 changed files with 13 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
main.py
View File

@ -292,11 +292,15 @@ def main():
elif args.command == 'add': elif args.command == 'add':
playlist = get_playlist_id(args.playlist) playlist = get_playlist_id(args.playlist)
pl_videos = {video['snippet']['resourceId']['videoId'] # {video_id: video_title} for videos already in the playlist
for video in list_playlist(youtube, playlist)} pl_videos = {pl_item['snippet']['resourceId']['videoId']: pl_item['snippet']['title']
for pl_item in list_playlist(youtube, playlist)}
for video_id in args.video_ids: for video_id in args.video_ids:
if video_id in pl_videos: if video_id in pl_videos:
print(f'Video {video_id} already in playlist {args.playlist}') short_title = _truncate_title(pl_videos[video_id])
print(f"Video '{short_title}' [{video_id}]"
f" is already in playlist {args.playlist}")
continue continue
add_video_to_playlist(youtube, video_id, playlist, args.dry_run) add_video_to_playlist(youtube, video_id, playlist, args.dry_run)
@ -307,13 +311,16 @@ def main():
next(reader, None) # skip the headers next(reader, None) # skip the headers
video_ids.extend(row[0] for row in reader if row) video_ids.extend(row[0] for row in reader if row)
playlist = get_playlist_id(args.playlist) playlist = get_playlist_id(args.playlist)
pl_videos = {video['snippet']['resourceId']['videoId'] # {video_id: video_title} for videos already in the playlist
for video in list_playlist(youtube, playlist)} pl_videos = {pl_item['snippet']['resourceId']['videoId']: pl_item['snippet']['title']
for pl_item in list_playlist(youtube, playlist)}
if args.limit > 0: if args.limit > 0:
video_ids = video_ids[:args.limit] video_ids = video_ids[:args.limit]
for video_id in video_ids: for video_id in video_ids:
if video_id in pl_videos: if video_id in pl_videos:
print(f'Video {video_id} already in playlist {args.playlist}') short_title = _truncate_title(pl_videos[video_id])
print(f"Video '{short_title}' [{video_id}]"
f" is already in playlist {args.playlist}")
continue continue
add_video_to_playlist(youtube, video_id, playlist, args.dry_run) add_video_to_playlist(youtube, video_id, playlist, args.dry_run)