diff --git a/spotify-backup.py b/spotify-backup.py new file mode 100644 index 0000000..3f84dd6 --- /dev/null +++ b/spotify-backup.py @@ -0,0 +1,125 @@ +import argparse +import urllib +import urllib2 +import json +import time +import webbrowser +import sys +import BaseHTTPServer +import re +import codecs + +class SpotifyAPI: + def __init__(self, auth): + self._auth = auth + + def get(self, url, params={}, tries=3): + if not url.startswith('https://api.spotify.com/v1/'): + url = 'https://api.spotify.com/v1/' + url + if params: + url += ('&' if '?' in url else '?') + urllib.urlencode(params) + + try: + return json.load(urllib2.urlopen(urllib2.Request(url, None, {'Authorization': 'Bearer ' + self._auth}))) + except urllib2.HTTPError as err: + log('Couldn\'t load URL: {} ({} {})'.format(url, err.code, err.reason)) + if tries <= 0: + sys.exit(1) + time.sleep(2) + log('Trying again...') + return self.get(url, tries=tries-1) + + def list(self, path, params={}): + response = self.get(path, params) + items = response['items'] + while response['next']: + response = self.get(response['next']) + items += response['items'] + return items + + _LISTEN_PORT_NUMBER = 43019 + + @staticmethod + def authorize(client_id, scope): + webbrowser.open('https://accounts.spotify.com/authorize?' + urllib.urlencode({ + 'response_type': 'token', + 'client_id': client_id, + 'scope': scope, + 'redirect_uri': 'http://127.0.0.1:{}/redirect'.format(SpotifyAPI._LISTEN_PORT_NUMBER) + })) + httpd = SpotifyAPI._AuthorizationListener('', SpotifyAPI._LISTEN_PORT_NUMBER) + try: + while True: + httpd.handle_request() + except SpotifyAPI._Authorization as auth: + return SpotifyAPI(auth.access_token) + + class _AuthorizationListener(BaseHTTPServer.HTTPServer): + def __init__(self, host, port): + BaseHTTPServer.HTTPServer.__init__(self, (host, port), SpotifyAPI._AuthorizationHandler) + def handle_error(self, request, client_address): + raise + + class _AuthorizationHandler(BaseHTTPServer.BaseHTTPRequestHandler): + def do_GET(self): + if self.path.startswith('/redirect'): + self.send_response(200) + self.send_header('Content-Type', 'text/html') + self.end_headers() + self.wfile.write('') + elif self.path.startswith('/token?'): + self.send_response(200) + self.send_header('Content-Type', 'text/html') + self.end_headers() + self.wfile.write('Thanks! You may now close this window.') + raise SpotifyAPI._Authorization(re.search('access_token=([^&]*)', self.path).group(1)) + else: + self.send_error(404) + + def log_message(self, format, *args): + pass + + class _Authorization(Exception): + def __init__(self, access_token): + self.access_token = access_token + +def log(str): + print u'[{}] {}'.format(time.strftime('%I:%M:%S'), str).encode(sys.stdout.encoding, errors='replace') + +def main(): + parser = argparse.ArgumentParser(description='Exports your Spotify playlists. By default, opens a browser window to authorize the Spotify Web API, but you can manually specify an OAuth token with the --token option.') + parser.add_argument('--token', metavar='OAUTH_TOKEN', help='use a Spotify OAuth token (requires the `playlist-read-private` permission)') + parser.add_argument('--format', default='txt', choices=['json', 'txt'], help='output format (default: txt)') + parser.add_argument('file', help='output filename') + args = parser.parse_args() + + if args.token: + spotify = SpotifyAPI(args.token) + else: + spotify = SpotifyAPI.authorize(client_id='5c098bcc800e45d49e476265bc9b6934', scope='playlist-read-private') + + me = spotify.get('me') + log(u'Logged in as {display_name} ({id})'.format(**me)) + playlists = spotify.list('users/{user_id}/playlists'.format(user_id=me['id']), {'limit': 50}) + for playlist in playlists: + log(u'Loading playlist: {name} ({tracks[total]} songs)'.format(**playlist)) + playlist['tracks'] = spotify.list(playlist['tracks']['href'], {'limit': 100}) + + with codecs.open(args.file, 'w', 'utf-8') as f: + if args.format == 'json': + json.dump(playlists, f) + elif args.format == 'txt': + for playlist in playlists: + f.write(playlist['name'] + '\r\n') + for track in playlist['tracks']: + f.write(u'{uri}\t{name}\t{artists}\t{album}\r\n'.format( + uri=track['track']['uri'], + name=track['track']['name'], + artists=', '.join([artist['name'] for artist in track['track']['artists']]), + album=track['track']['album']['name'] + )) + f.write('\r\n') + log('Wrote file: ' + args.file) + +if __name__ == '__main__': + main() \ No newline at end of file