def start_client():
self.client = SpotifyClient(username, password)
self.client.start()
return self.client.login_dfr
class SpotifyResourceProvider(ResourceProvider):
"""
Manages the pyspotify session
"""
supported_uri = 'spotify://.*'
default_config = {'username': '', 'password': ''}
def __init__(self, *args, **kwargs):
Loggable.__init__(self, *args, **kwargs)
super(SpotifyResourceProvider, self).__init__(*args, **kwargs)
self.debug("spotify resource privoder is now initialized")
def initialize(self):
self.client = None
username = self.config['username'] or None
password = self.config['password'] or None
if username and password:
self.debug("logging in from initialize")
self.login(username, password)
dfr = super(SpotifyResourceProvider, self).initialize()
return dfr
def login(self, username, password):
"""Log in by creating and starting a new spotify client"""
def start_client():
self.client = SpotifyClient(username, password)
self.client.start()
return self.client.login_dfr
if self.client:
dfr = self.logout()
dfr.addCallback(start_client)
return dfr
else:
return start_client()
def logout(self):
"""Disconnect the client and return a deferred fired on logout"""
self.client.disconnect()
self.client = None
return self.client.logout_dfr
def is_logged_in(self):
return self.client is not None
def clean(self):
"""Makes sure we disconnect properly before shutdown"""
self.debug("Running spotify resource provider clean method")
self.client.disconnect()
dfr = self.client.logout_dfr
dfr.addCallback(lambda r: super(SpotifyResourceProvider, self).clean())
return dfr
def get_playlists(self):
"""
Return playlists as model instances.
TODO: Create the track models when the playlist is loaded
instead.
"""
playlists = []
for playlist in self.client.playlists:
playlist_model = SpotifyPlaylistModel()
playlist_model.title = playlist.name()
playlist_model.uri = unicode(pyspotify.Link.from_playlist(playlist))
for track in playlist:
track_model = SpotifyTrackModel()
track_model.title = track.name()
track_model.artists = map(unicode, track.artists())
track_model.artist = ', '.join(track_model.artists)
track_model.album = unicode(track.album())
track_model.playable_uri = str(pyspotify.Link.from_track(track, 0))
playlist_model.append(track_model)
playlists.append(playlist_model)
return playlists
def get(self, uri, context_model=None):
"""Handle spotify:// URIs"""
if uri == 'spotify://playlists':
return threads.deferToThread(self.get_playlists)
