Address comments

This commit is contained in:
Alok Saboo 2022-06-13 09:26:15 -04:00
parent 4bb8862b6f
commit a2e6680e2f

View file

@ -32,6 +32,7 @@ from beets import ui
from beets.autotag.hooks import AlbumInfo, TrackInfo from beets.autotag.hooks import AlbumInfo, TrackInfo
from beets.plugins import MetadataSourcePlugin, BeetsPlugin from beets.plugins import MetadataSourcePlugin, BeetsPlugin
DEFAULT_WAITING_TIME = 5
class SpotifyPlugin(MetadataSourcePlugin, BeetsPlugin): class SpotifyPlugin(MetadataSourcePlugin, BeetsPlugin):
data_source = 'Spotify' data_source = 'Spotify'
@ -165,10 +166,11 @@ class SpotifyPlugin(MetadataSourcePlugin, BeetsPlugin):
self._authenticate() self._authenticate()
return self._handle_response(request_type, url, params=params) return self._handle_response(request_type, url, params=params)
elif response.status_code == 429: elif response.status_code == 429:
seconds = response.headers['Retry-After'] seconds = response.headers.get('Retry-After',
time.sleep(int(seconds)) DEFAULT_WAITING_TIME)
self._log.info('Too many API requests. Retrying after {} \ self._log.debug('Too many API requests. Retrying after {} \
seconds.', seconds) seconds.', seconds)
time.sleep(int(seconds))
return self._handle_response(request_type, url, params=params) return self._handle_response(request_type, url, params=params)
else: else:
raise ui.UserError( raise ui.UserError(