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

@@ -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

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,