added path specification for export and user playlists only
This commit is contained in:
parent
447f68a893
commit
45de569308
@ -2,11 +2,14 @@ import spotframework.net.user as userclass
|
||||
import spotframework.net.playlist as playlist
|
||||
import spotframework.io.csv as csvwrite
|
||||
|
||||
import sys
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
||||
user = userclass.User()
|
||||
|
||||
playlists = playlist.getPlaylists(user)
|
||||
playlists = playlist.getUserPlaylists(user)
|
||||
|
||||
for play in playlists:
|
||||
csvwrite.exportPlaylist(user, play['id'], play['name'])
|
||||
csvwrite.exportPlaylist(user, play['id'], play['name'], sys.argv[1])
|
||||
print(play['name'] + ' exported')
|
||||
|
2
main.py
2
main.py
@ -27,6 +27,6 @@ if __name__ == '__main__':
|
||||
|
||||
import spotframework.io.csv as csvwrite
|
||||
|
||||
csvwrite.exportPlaylist(user, '000Eh2vXzYGgrEFlgcWZj3', 'february')
|
||||
csvwrite.exportPlaylist(user, '000Eh2vXzYGgrEFlgcWZj3', 'february', '')
|
||||
|
||||
print(user.access_token)
|
||||
|
@ -4,11 +4,11 @@ import datetime
|
||||
|
||||
headers = ['name', 'artist', 'album', 'album artist', 'added', 'spotify id', 'added by']
|
||||
|
||||
def exportPlaylist(user, playlistid, name):
|
||||
def exportPlaylist(user, playlistid, name, path):
|
||||
|
||||
playlist = playlistpull.getPlaylistTracks(user, playlistid)
|
||||
|
||||
with open('{}_{}.csv'.format(name, str(datetime.datetime.now()).split('.')[0]), 'w') as fileobj:
|
||||
with open('{}{}_{}.csv'.format(path, name.replace('/', '_'), str(datetime.datetime.now()).split('.')[0]), 'w') as fileobj:
|
||||
|
||||
writer = csv.DictWriter(fileobj, fieldnames = headers)
|
||||
writer.writeheader()
|
||||
|
@ -57,7 +57,7 @@ def getPlaylistTracks(user, playlistid, offset = 0):
|
||||
|
||||
if req.status_code == 200:
|
||||
|
||||
print(req.text)
|
||||
#print(req.text)
|
||||
resp = req.json()
|
||||
|
||||
tracks += resp['items']
|
||||
@ -70,6 +70,4 @@ def getPlaylistTracks(user, playlistid, offset = 0):
|
||||
return tracks
|
||||
|
||||
else:
|
||||
return None
|
||||
|
||||
|
||||
raise ValueError("Couldn't Pull Playlist " + str(playlistid) + ' ' + str(req.status_code))
|
||||
|
Loading…
Reference in New Issue
Block a user