added catch for no network username

This commit is contained in:
aj 2019-10-10 21:56:50 +01:00
parent 25e6998a79
commit f868d1425a
2 changed files with 22 additions and 19 deletions

View File

@ -27,28 +27,28 @@ if __name__ == '__main__':
logger.addHandler(stream_handler) logger.addHandler(stream_handler)
try: # try:
network = Network(NetworkUser(os.environ['SPOTCLIENT'], network = Network(NetworkUser(os.environ['SPOTCLIENT'],
os.environ['SPOTSECRET'], os.environ['SPOTSECRET'],
os.environ['SPOTREFRESH'])) os.environ['SPOTREFRESH']))
network.user.refresh_access_token() network.user.refresh_access_token()
playlists = network.get_user_playlists() playlists = network.get_user_playlists()
for playlist in playlists: for playlist in playlists:
playlist.tracks = network.get_playlist_tracks(playlist.uri) playlist.tracks = network.get_playlist_tracks(playlist.uri)
path = sys.argv[1] path = sys.argv[1]
datepath = str(datetime.datetime.now()).split(' ')[0].replace('-', '/') datepath = str(datetime.datetime.now()).split(' ')[0].replace('-', '/')
totalpath = os.path.join(path, datepath) totalpath = os.path.join(path, datepath)
pathdir = os.path.dirname(totalpath) pathdir = os.path.dirname(totalpath)
if not os.path.exists(totalpath): if not os.path.exists(totalpath):
os.makedirs(totalpath) os.makedirs(totalpath)
for play in playlists: for play in playlists:
csvwrite.export_playlist(play, totalpath) csvwrite.export_playlist(play, totalpath)
except Exception as e: # except Exception as e:
logger.exception(f'exception occured') # logger.exception(f'exception occured')

View File

@ -284,7 +284,10 @@ class Network:
playlists = self.get_playlists() playlists = self.get_playlists()
if playlists: if self.user.username is None:
self.user.refresh_info()
if playlists is not None:
return list(filter(lambda x: x.owner.username == self.user.username, playlists)) return list(filter(lambda x: x.owner.username == self.user.username, playlists))
else: else:
logger.error('no playlists returned to filter') logger.error('no playlists returned to filter')