From f7961bf6a391c5b195233a23764a7544d2d285ac Mon Sep 17 00:00:00 2001 From: Aaditya Dhruv Date: Thu, 5 Jun 2025 16:46:09 -0500 Subject: Add single album support --- yt_music_scraper/main.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/yt_music_scraper/main.py b/yt_music_scraper/main.py index 014fe11..913e16a 100644 --- a/yt_music_scraper/main.py +++ b/yt_music_scraper/main.py @@ -82,6 +82,7 @@ def download_album(album_id): if __name__ == "__main__": parser = argparse.ArgumentParser(prog="YT DLP Music downloader", description="Download all albums corresponding to songs in a playlist") parser.add_argument("playlist") + parser.add_argument("--single", action="store_true", default=False) args = parser.parse_args() @@ -92,7 +93,10 @@ if __name__ == "__main__": i = 0 if data: albums = set() - for datum in data["entries"]: + + entries = data["entries"] if not args.single else [data] + logger.debug(json.dumps(entries, indent=4)) + for datum in entries: if datum: logger.debug(json.dumps(datum, indent=4)) info = vid_info(datum) -- cgit