From 4328dd17bfcd4c33a77c4405d52c3660a6c62bf5 Mon Sep 17 00:00:00 2001 From: nil0j Date: Fri, 27 Sep 2024 17:27:02 +0200 Subject: update --- rust/scripts/populate/main.py | 46 +++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 46 insertions(+) create mode 100644 rust/scripts/populate/main.py (limited to 'rust/scripts/populate/main.py') diff --git a/rust/scripts/populate/main.py b/rust/scripts/populate/main.py new file mode 100644 index 0000000..ec0157c --- /dev/null +++ b/rust/scripts/populate/main.py @@ -0,0 +1,46 @@ +from albums import album_data +import api +import parser +import database + + +def start(): + print("downloading data...") + api.download_albums(album_data) + + print("uploading data...") + upload_albums( + get_album_data(), + get_artist_names() + ) + database.close() + print("upload finished!") + print("remember to move the covers directory once you're done") + + +def upload_albums(album_data, artist_names): + database.process_artists(artist_names) + database.process_albums(album_data) + + +def get_album_data(): + result = [] + + for artist_id, artist in enumerate(album_data, 1): + album_list = artist[1] + + for album_id, album in enumerate(album_list, 1): + result.append(parser.process_json_file(album, album_id, artist_id)) + + return result + + +def get_artist_names(): + artist_data = [] + [artist_data.append( + artist[0] + ) for artist in album_data] + return artist_data + + +start() -- cgit v1.2.3