Add get_library

1 files changed, 131 insertions, 0 deletions

diff --git a/get_library b/get_library
new file mode 100755
index 0000000..db7e6ec
--- /dev/null
+++ b/get_library
@@ -0,0 +1,131 @@
+#!/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)
+
+