Fix album name retrieval in get_artist_tracks function

ChocolateApp · Jan 8, 2024 · eaada51 · eaada51
1 parent 0f6c8fe
commit eaada51
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/chocolate_app/__main__.py b/src/chocolate_app/__main__.py
@@ -1580,7 +1580,11 @@ def get_artist_tracks(artist_id):
     for track in tracks_list:
         del track["_sa_instance_state"]
         try:
-            album_name = Albums.query.filter_by(id=track["album_id"]).first().name
+            album_name = Albums.query.filter_by(id=track["album_id"]).first()
+            if album_name:
+                album_name = album_name.name
+            else:
+                album_name = "Track"
             track["album_name"] = album_name
         except Exception as e:
             log_message = f"Error while getting album name for track {track['id']}: {e}"