From cbe23bd0f77e5f82019f311f45c32381652f1a9e Mon Sep 17 00:00:00 2001 From: Maks Snegov Date: Wed, 7 Feb 2024 12:34:17 -0800 Subject: [PATCH] Fix limits for add-csv option --- robocyp.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/robocyp.py b/robocyp.py index ada973c..90646c7 100755 --- a/robocyp.py +++ b/robocyp.py @@ -329,15 +329,16 @@ def main(): # {video_id: video_title} for videos already in the playlist pl_videos = {pl_item['snippet']['resourceId']['videoId']: pl_item['snippet']['title'] for pl_item in list_playlist(youtube, playlist_id)} - if args.limit > 0: - video_ids = video_ids[:args.limit] + processed = 0 for video_id in video_ids: + if 0 <= args.limit <= processed: + break if video_id in pl_videos: short_title = _truncate_title(pl_videos[video_id]) print(f"Video '{short_title}' [{video_id}]" f" is already in playlist {args.playlist}") continue - add_video_to_playlist(youtube, video_id, playlist_id, args.dry_run) + processed += int(add_video_to_playlist(youtube, video_id, playlist_id, args.dry_run)) elif args.command == "dups": processed = 0