diff options
| -rw-r--r-- | yt_music_scraper/main.py | 17 |
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.*") |