first commit

swingmusic/setup/__init__.py

@@ -0,0 +1,55 @@
+"""
+Prepares the server for use.
+"""
+
+import uuid
+from dataclasses import asdict
+from time import time
+
+from swingmusic.config import UserConfig
+from swingmusic.lib.mapstuff import (
+    map_album_colors,
+    map_artist_colors,
+    map_favorites,
+    map_scrobble_data,
+)
+from swingmusic.setup.sqlite import run_migrations, setup_sqlite
+from swingmusic.store.albums import AlbumStore
+from swingmusic.store.artists import ArtistStore
+from swingmusic.store.folder import FolderStore
+from swingmusic.store.tracks import TrackStore
+from swingmusic.utils.generators import get_random_str
+
+
+def run_setup():
+    """
+    Creates the config directory, runs migrations, and loads settings.
+    """
+
+    # setup config file
+    config = UserConfig()
+    config.setup_config_file()
+
+    if not config.serverId:
+        config.serverId = str(uuid.uuid4())
+        config.write_to_file(asdict(config))
+
+    setup_sqlite()
+    run_migrations()
+
+
+def load_into_mem():
+    """
+    Load all tracks, albums, and artists into memory.
+    """
+    # INFO: Load all tracks, albums, and artists data into memory
+    key = str(time())
+    TrackStore.load_all_tracks(get_random_str())
+    AlbumStore.load_albums(key)
+    ArtistStore.load_artists(key)
+    FolderStore.load_filepaths()
+
+    map_scrobble_data()
+    map_favorites()
+    map_artist_colors()
+    map_album_colors()