Add artists to search to narrow down album

1 files changed, 14 insertions, 3 deletions

diff --git a/yt_music_scraper/main.py b/yt_music_scraper/main.py
index 0a0ec72..849c6ef 100644
--- a/yt_music_scraper/main.py
+++ b/yt_music_scraper/main.py
@@ -20,11 +20,18 @@ def album_info(data):
     logger.info(json.dumps(data['playlist_id'] if 'playlist_id' in data else "N/A", indent=4))
     return data['playlist_id'] if 'playlist_id' in data else "N/A"
 
+
+
 def vid_info(data):
 
     logger.info(json.dumps(data['title'] if 'title' in data else "N/A", indent=4))
     logger.info(json.dumps(data['album'] if 'album' in data else "N/A", indent=4))
-    return data['album'] if 'album' in data else "N/A"
+    album = data['album'] if 'album' in data else None
+    artists = ','.join(data['artists']) if 'artists' in data else ""
+    if album:
+        return album + " " +  artists 
+    else:
+        return None
 
 ydl_opts2 = {
     'format': 'bestaudio/best',
@@ -86,8 +93,12 @@ if __name__ == "__main__":
         for datum in data["entries"]:
             if datum:
                 logger.debug(json.dumps(datum, indent=4))
-                albums.add(vid_info(datum))
-                logger.info(f"ALBUM NAME: {vid_info(datum)}")
+                info = vid_info(datum)
+                if info:
+                    albums.add(info)
+                    logger.info(f"ALBUM NAME: {info}")
+                else:
+                    logger.info(f"Skipping {datum}")
 
         logger.info(f"albums: {albums}")
         re = re.compile("OLAK.*")