diff options
| -rwxr-xr-x | get_library | 131 |
1 files changed, 0 insertions, 131 deletions
diff --git a/get_library b/get_library deleted file mode 100755 index db7e6ec..0000000 --- a/get_library +++ /dev/null @@ -1,131 +0,0 @@ -#!/usr/bin/env python3 -import json -# Can't remember what I needed this for... Maybe checking if song already -# exists? -# import musicpd -import music_tag -import os -import youtube_dl -from ytmusicapi import YTMusic - -DEBUG = True - -# Includes the format characters -YOUTUBE_MUSIC_URL = "https://music.youtube.com/watch?v={}" - -""" -How the albums will be formatted -{ - 'id': { - 'name': '', - 'tracks': [ - { - 'id': '', - 'name': '' - } - ], - "thumbnail_url": '', - "is_thumbnail_downloaded": False, - "thumbnail_filename": '', - } -} -""" -ALBUMS = {} - - -def create_downloader(music_directory, cookies): - audio_options = { - 'format': 'mp3/bestaudio/best', - 'cookiefile': cookies, - 'outtmpl': os.path.join(music_directory, '%(title)s.%(ext)s'), - 'postprocessors': [ - { - 'key': 'FFmpegExtractAudio', - 'preferredcodec': 'mp3', - 'preferredquality': '192', - }, - {'key': 'FFmpegMetadata'}, - ], - 'writeinfojson': True, - 'quiet': not DEBUG - } - -def write_metadata_to_song_file(filename, metadata): - file = music_tag.load_file(filename) - - file['name'] = metadata['title'] - file['artist'] = metadata['artist'] - file['album'] = metadata['album'] - file['year'] = metadata['year'] - - file.save() - -if __name__ == "__main__": - app = YTMusic('../headers_auth.json') - song_limit = 25 - - all_raw_song_data = app.get_library_songs(limit=song_limit, order='a_to_z') - - if DEBUG: - with open('raw_song_data.json', 'w') as f: - f.write(json.dumps(all_raw_song_data)) - - all_parsed_songs = [] - for raw_song in all_raw_song_data: - song_id = raw_song['videoId'] - album_id = raw_song['album']['id'] - if album_id not in ALBUMS.keys(): - raw_album = app.get_album(album_id) - ALBUMS[album_id] = { - 'name': raw_album['title'], - 'tracks': [{'title': song['title'], 'id': song['videoId'], 'track_num': index + 1} for index, song in enumerate(raw_album['tracks'])], - 'year': raw_album['year'], - # TODO: Add thumbnail_url - } - - album = ALBUMS[album_id] - - track_num = None - for track in ALBUMS[album_id]['tracks']: - if track['id'] == song_id: - track_num = track['track_num'] - - parsed_song = { - 'id': raw_song['videoId'], - 'title': raw_song['title'], - 'artists': [artist['name'] for artist in raw_song['artists']], - 'album': raw_song['album']['name'], - 'track': track_num, - 'url': YOUTUBE_MUSIC_URL.format(raw_song['videoId']), - 'year': album['year'] - } - all_parsed_songs.append(parsed_song) - - if DEBUG: - with open('parsed_album_data.json', 'w') as f: - f.write(json.dumps({'albums': ALBUMS})) - - with open('parsed_song_data.json', 'w') as f: - f.write(json.dumps({'songs': all_parsed_songs})) - - - # music_directory = os.path.join(os.path.expanduser("~"), "Music") - music_directory = './music' - - # TODO: Can you use the cookies from YTMusic here - # cookies = os.path.join(os.path.expanduser("~"), "cookies.txt") - cookies = './cookies.txt' - ytdl = create_downloader(music_directory, cookies) - - failed_songs = [] - for song in all_parsed_songs: - try: - ytdl.extract_info(song['url'], download=True) - except Exception as ex: - print(ex) - print("Could not download: {}".format((song))) - failed_songs.append(song) - - files = get_all_files(music_directory) - - |