From e0d5de4714030d6ed55cb3237b1c94eb587f63fa Mon Sep 17 00:00:00 2001
From: Alok Saboo <arsaboo@gmx.com>
Date: Tue, 5 Jul 2022 20:58:02 -0400
Subject: [PATCH] Add album information to spotify tracks

update

Update spotify.py

Update spotify.py

Update spotify.py

Update spotify.py

Update spotify.py

Update spotify.py

Update spotify.py

Update spotify.py

Update spotify.py

Update hooks.py

Update hooks.py

Update spotify.py

Update spotify.py

Update spotify.py

Update spotify.py

Update spotify.py

Update spotify.py

Update hooks.py

Update spotify.py

Update spotify.py

Cleanup

more cleanup
---
 beets/autotag/hooks.py | 3 ++-
 beetsplug/spotify.py   | 8 ++++++++
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/beets/autotag/hooks.py b/beets/autotag/hooks.py
index 30904ff29..070c98abc 100644
--- a/beets/autotag/hooks.py
+++ b/beets/autotag/hooks.py
@@ -160,7 +160,7 @@ class TrackInfo(AttrDict):
                  artist_sort=None, disctitle=None, artist_credit=None,
                  data_source=None, data_url=None, media=None, lyricist=None,
                  composer=None, composer_sort=None, arranger=None,
-                 track_alt=None, work=None, mb_workid=None,
+                 track_alt=None, work=None, mb_workid=None,album=None,
                  work_disambig=None, bpm=None, initial_key=None, genre=None,
                  **kwargs):
         self.title = title
@@ -172,6 +172,7 @@ class TrackInfo(AttrDict):
         self.index = index
         self.media = media
         self.medium = medium
+        self.album=album
         self.medium_index = medium_index
         self.medium_total = medium_total
         self.artist_sort = artist_sort
diff --git a/beetsplug/spotify.py b/beetsplug/spotify.py
index 75ae52891..a692accc4 100644
--- a/beetsplug/spotify.py
+++ b/beetsplug/spotify.py
@@ -291,11 +291,19 @@ class SpotifyPlugin(MetadataSourcePlugin, BeetsPlugin):
         :rtype: beets.autotag.hooks.TrackInfo
         """
         artist, artist_id = self.get_artist(track_data['artists'])
+
+        # Get album information for spotify tracks
+        try:
+            album=track_data['album']['name']
+        except KeyError:
+            album=None
+            pass
         return TrackInfo(
             title=track_data['name'],
             track_id=track_data['id'],
             spotify_track_id=track_data['id'],
             artist=artist,
+            album=album,
             artist_id=artist_id,
             spotify_artist_id=artist_id,
             length=track_data['duration_ms'] / 1000,