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,