Added 'Liked' songs capability with new 'scope' switch.

pull/21/head
lucas_a_morton 2020-09-06 21:07:52 -07:00
parent 2c23bc0604
commit d8df1f79fe
1 changed files with 10 additions and 6 deletions

View File

@ -123,7 +123,7 @@ def main():
parser.add_argument('--token', metavar='OAUTH_TOKEN', help='use a Spotify OAuth token (requires the ' parser.add_argument('--token', metavar='OAUTH_TOKEN', help='use a Spotify OAuth token (requires the '
+ '`playlist-read-private` permission)') + '`playlist-read-private` permission)')
parser.add_argument('--format', default='txt', choices=['json', 'txt'], help='output format (default: txt)') parser.add_argument('--format', default='txt', choices=['json', 'txt'], help='output format (default: txt)')
parser.add_argument('--scope', default='playlist-read-collaborative', choices=['playlist-read-private', 'playlist-read-collaborative'], help='Spotify Scope to use, to get private or private and collaborative lists. (default: playlist-read-collaborative)') parser.add_argument('--scope', default='playlist-read-private', choices=['playlist-read-private', 'playlist-read-collaborative','user-library-read'], help='Spotify Scope to use, to get private or private and collaborative lists. (default: playlist-read-collaborative)')
parser.add_argument('file', help='output filename', nargs='?') parser.add_argument('file', help='output filename', nargs='?')
args = parser.parse_args() args = parser.parse_args()
@ -142,11 +142,15 @@ def main():
log('Logged in as {display_name} ({id})'.format(**me)) log('Logged in as {display_name} ({id})'.format(**me))
# List all playlists and all track in each playlist. # List all playlists and all track in each playlist.
playlists = spotify.list('users/{user_id}/playlists'.format(user_id=me['id']), {'limit': 50}) if args.scope in ['playlist-read-private', 'playlist-read-collaborative']:
for playlist in playlists: playlists = spotify.list('users/{user_id}/playlists'.format(user_id=me['id']), {'limit': 50})
log('Loading playlist: {name} ({tracks[total]} songs)'.format(**playlist)) for playlist in playlists:
playlist['tracks'] = spotify.list(playlist['tracks']['href'], {'limit': 100}) log('Loading playlist: {name} ({tracks[total]} songs)'.format(**playlist))
playlist['tracks'] = spotify.list(playlist['tracks']['href'], {'limit': 100})
elif args.scope == 'user-library-read': #Get the 'Liked'/'Saved' tracks
likedlist = spotify.list('users/{user_id}/tracks'.format(user_id=me['id']), {'limit': 50})
likedlist_dict = {'name':'Liked Songs','tracks':likedlist}
playlists = [likedlist_dict]
# Write the file. # Write the file.
with open(args.file, 'w', encoding='utf-8') as f: with open(args.file, 'w', encoding='utf-8') as f:
# JSON file. # JSON file.