mirror of
https://github.com/caseychu/spotify-backup.git
synced 2024-05-11 10:51:31 +00:00
73 lines
2.5 KiB
Python
73 lines
2.5 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import argparse
|
|
import codecs
|
|
import http.client
|
|
import http.server
|
|
import json
|
|
import re
|
|
import sys
|
|
import time
|
|
import urllib.error
|
|
import urllib.parse
|
|
import urllib.request
|
|
import webbrowser
|
|
|
|
import importlib
|
|
spotifyBackup = importlib.import_module('spotify-backup')
|
|
SpotifyAPI = spotifyBackup.SpotifyAPI
|
|
log = spotifyBackup.log
|
|
|
|
def main():
|
|
# Parse arguments.
|
|
parser = argparse.ArgumentParser(description='Exports your Spotify favourite songs. By default, opens a browser window '
|
|
+ 'to authorize the Spotify Web API, but you can also manually specify'
|
|
+ ' an OAuth token with the --token option.')
|
|
parser.add_argument('--token', metavar='OAUTH_TOKEN', help='use a Spotify OAuth token (requires the '
|
|
+ '`user-library-read` permission)')
|
|
parser.add_argument('--format', default='txt', choices=['json', 'txt'], help='output format (default: txt)')
|
|
parser.add_argument('--scope', default='user-library-read', choices=['user-library-read'], help='Spotify Scope to use, to get favorite songs. (default: user-library-read)')
|
|
parser.add_argument('file', help='output filename', nargs='?')
|
|
args = parser.parse_args()
|
|
|
|
# If they didn't give a filename, then just prompt them. (They probably just double-clicked.)
|
|
while not args.file:
|
|
args.file = input('Enter a file name (e.g. playlists.txt): ')
|
|
args.format = args.file.split('.')[-1]
|
|
|
|
# Log into the Spotify API.
|
|
if args.token:
|
|
spotify = SpotifyAPI(args.token)
|
|
else:
|
|
spotify = SpotifyAPI.authorize(client_id='5c098bcc800e45d49e476265bc9b6934', scope=args.scope)
|
|
|
|
# Get the ID of the logged in user.
|
|
me = spotify.get('me')
|
|
log('Logged in as {display_name} ({id})'.format(**me))
|
|
|
|
# List all favorite songs
|
|
log('Loading favourite songs')
|
|
favourites = spotify.list('me/tracks', {'limit': 50})
|
|
|
|
# Write the file.
|
|
log('Writing files...')
|
|
with open(args.file, 'w', encoding='utf-8') as f:
|
|
# JSON file.
|
|
if args.format == 'json':
|
|
json.dump(favourites, f)
|
|
|
|
# Tab-separated file.
|
|
elif args.format == 'txt':
|
|
for track in favourites:
|
|
f.write('{name}\t{artists}\t{album}\t{uri}\t{added_at}\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'],
|
|
added_at=track['added_at']
|
|
))
|
|
log('Wrote file: ' + args.file)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|