Add single album supportHEADmaster

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)