From 3048638d9b423ed7b56d987ea6f74367d18a7c9d Mon Sep 17 00:00:00 2001
From: Broque Thomas <batmanesignal@gmail.com>
Date: Thu, 19 Feb 2026 22:36:57 -0800
Subject: [PATCH] Preserve full album context when adding tracks to wishlist

---
 web_server.py | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

diff --git a/web_server.py b/web_server.py
index d9af2677..c749529e 100644
--- a/web_server.py
+++ b/web_server.py
@@ -12652,15 +12652,13 @@ def _ensure_spotify_track_format(track_info):
         else:
             artists_list.append({'name': 'Unknown Artist'})
     
-    # Build album object with images if available
+    # Build album object — preserve ALL fields (id, release_date, total_tracks,
+    # album_type, images, etc.) so wishlist tracks retain full album context
+    # for correct folder placement, multi-disc support, and classification
     album_data = track_info.get('album', {})
     if isinstance(album_data, dict):
-        album = {
-            'name': album_data.get('name', 'Unknown Album')
-        }
-        # Preserve album images if present (important for ListenBrainz tracks)
-        if 'images' in album_data:
-            album['images'] = album_data['images']
+        album = dict(album_data)  # Copy all fields
+        album.setdefault('name', 'Unknown Album')
     else:
         album = {
             'name': str(album_data) if album_data else 'Unknown Album'