Populate album_genres on playlist manager page load

Author: CalPinSW

backend/src/controllers/database.py

@@ -140,7 +140,7 @@ def populate_album_genres_by_user_id(
     for idx, db_album in enumerate(albums):
         print("\033[A                             \033[A")
         print(f"processing album {idx} of {len(albums)}, skipped {skip_count}")
-        if get_album_genres(db_album) != []:
+        if get_album_genres(db_album.id) != []:
             skip_count += 1
             continue
         album_artists = get_album_artists(db_album)

backend/src/database/crud/album.py

@@ -59,12 +59,12 @@ def update_album(album: Album):
     return album
 
 
-def get_album_genres(album: DbAlbum) -> List[str]:
+def get_album_genres(album_id: str) -> List[str]:
     query = (
         DbGenre.select()
         .join(AlbumGenreRelationship)
         .join(DbAlbum)
-        .where(DbAlbum.id == album.id)
+        .where(DbAlbum.id == album_id)
     )
     return list(query)
 

backend/src/spotify.py

@@ -4,6 +4,7 @@
 import urllib.parse
 from typing import List, Optional
 from flask import Response, make_response, redirect
+from src.database.crud.album import get_album_genres
 from src.dataclasses.album import Album
 from src.dataclasses.playback_info import PlaybackInfo, PlaylistProgression
 from src.dataclasses.playback_request import (
@@ -263,6 +264,8 @@ def get_playlist_album_info(self, access_token, id) -> List[Album]:
         for track in playlist_tracks:
             if track.track.album not in playlist_albums:
                 playlist_albums.append(track.track.album)
+        for album in playlist_albums:
+            album.genres = [genre.name for genre in get_album_genres(album.id)]
         return playlist_albums
 
     def get_playlist_tracks(self, access_token, id: str):