diff options
| -rw-r--r-- | yt_music_scraper/main.py | 6 |
1 files changed, 5 insertions, 1 deletions
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) |