def main(): urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning) os.chdir(sys.path[0]) # Get args args = cli.get_args() # Load config BRUTEFORCE = args.bruteforce or BRUTEFORCEREGION preset = PRESETS[args.preset] # Parse options preset['quality'] = [] preset['quality'].append('HI_RES') if preset['MQA_FLAC_24'] else None preset['quality'].append('LOSSLESS') if preset['FLAC_16'] else None preset['quality'].append('HIGH') if preset['AAC_320'] else None preset['quality'].append('LOW') if preset['AAC_96'] else None # Check for auth flag / session settings RSF = RedseaSessionFile('./config/sessions.pk') if args.urls[0] == 'auth' and len(args.urls) == 1: print('\nThe "auth" command provides the following methods:') print('\n list: Lists stored sessions if any exist') print(' add: Prompts for a TV or Mobile session. The TV option displays a 6 digit key which should be ' 'entered inside link.tidal.com where the user can login. The Mobile option prompts for a Tidal username ' 'and password. Both options authorize a session which then gets stored in the sessions file') print(' remove: Removes a stored session from the sessions file by name') print(' default: Set a default account for redsea to use when the -a flag has not been passed') print(' reauth: Reauthenticates with server to get new sessionId') print('\nUsage: redsea.py auth add\n') exit() elif args.urls[0] == 'auth' and len(args.urls) > 1: if args.urls[1] == 'list': RSF.list_sessions() exit() elif args.urls[1] == 'add': if len(args.urls) == 5: RSF.create_session(args.urls[2], args.urls[3], args.urls[4]) else: RSF.new_session() exit() elif args.urls[1] == 'remove': RSF.remove_session() exit() elif args.urls[1] == 'default': RSF.set_default() exit() elif args.urls[1] == 'reauth': RSF.reauth() exit() elif args.urls[0] == 'id': type = None md = MediaDownloader(TidalApi(RSF.load_session(args.account)), preset, Tagger(preset)) if len(args.urls) == 2: id = args.urls[1] if not id.isdigit(): # Check if id is playlist (UUIDv4) pattern = re.compile('^[0-9a-f]{8}-[0-9a-f]{4}-4[0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$') if pattern.match(id): try: result = md.playlist_from_id(id) type = 'p' except TidalError: print("The playlist id " + str(id) + " could not be found!") exit() else: print('The id ' + str(id) + ' is not valid.') exit() else: print('Example usage: python redsea.py id 92265335') exit() if type is None: type = md.type_from_id(id) if type: media_to_download = [{'id': id, 'type': type}] else: print("The id " + str(id) + " could not be found!") exit() elif args.urls[0] == 'explore': if len(args.urls) == 3: if args.urls[1] == 'dolby' and args.urls[2] == 'atmos': page = 'dolby_atmos' elif args.urls[1] == 'sony' and args.urls[2] == '360': page = '360' else: print("Example usage of explore: python redsea.py explore (dolby atmos/sony 360)") exit() else: print("Example usage of explore: python redsea.py explore (dolby atmos/sony 360)") exit() md = MediaDownloader(TidalApi(RSF.load_session(args.account)), preset, Tagger(preset)) page_content = md.page(page) show_more_link = page_content['rows'][len(page_content['rows'])-1]['modules'][0]['showMore']['apiPath'] now_available = md.page(show_more_link[6:]) items = now_available['rows'][0]['modules'][0]['pagedList']['items'] total_items = len(items) for i in range(total_items): item = items[i] if item['audioModes'] == ['DOLBY_ATMOS']: specialtag = " [Dolby Atmos]" elif item['audioModes'] == ['SONY_360RA']: specialtag = " [360 Reality Audio]" else: specialtag = "" if item['explicit']: explicittag = " [E]" else: explicittag = "" date = " (" + item['streamStartDate'].split('T')[0] + ")" print(str(i + 1) + ") " + str(item['title']) + " - " + str( item['artists'][0]['name']) + explicittag + specialtag + date) print(str(total_items+1) + ") Exit") while True: chosen = int(input("Album Selection: ")) - 1 if chosen == total_items: exit() elif chosen > total_items or chosen < 0: print("Enter an existing number") else: break print() media_to_download = [{'id': str(items[chosen]['id']), 'type': 'a'}] elif args.urls[0] == 'search': md = MediaDownloader(TidalApi(RSF.load_session(args.account)), preset, Tagger(preset)) while True: searchresult = md.search_for_id(args.urls[2:]) if args.urls[1] == 'track': searchtype = 'tracks' elif args.urls[1] == 'album': searchtype = 'albums' elif args.urls[1] == 'video': searchtype = 'videos' else: print("Example usage of search: python redsea.py search [track/album/video] Darkside Alan Walker") exit() # elif args.urls[1] == 'playlist': # searchtype = 'playlists' numberofsongs = int(searchresult[searchtype]['totalNumberOfItems']) if numberofsongs > 10: numberofsongs = 10 for i in range(numberofsongs): song = searchresult[searchtype]['items'][i] if searchtype != 'videos': if song['audioModes'] == ['DOLBY_ATMOS']: specialtag = " [Dolby Atmos]" elif song['audioModes'] == ['SONY_360RA']: specialtag = " [360 Reality Audio]" elif song['audioQuality'] == 'HI_RES': specialtag = " [MQA]" else: specialtag = "" else: specialtag = " [" + song['quality'].replace('MP4_', '') + "]" if song['explicit']: explicittag = " [E]" else: explicittag = "" print(str(i + 1) + ") " + str(song['title']) + " - " + str( song['artists'][0]['name']) + explicittag + specialtag) query = None if numberofsongs > 0: print(str(numberofsongs + 1) + ") Not found? Try a new search") while True: chosen = int(input("Song Selection: ")) - 1 if chosen == numberofsongs: query = input("Enter new search query: [track/album/video] Darkside Alan Walker: ") break elif chosen > numberofsongs: print("Enter an existing number") else: break print() if query: args.urls = ("search " + query).split() continue else: print("No results found for '" + ' '.join(args.urls[2:])) print("1) Not found? Try a new search") print("2) Quit") while True: chosen = int(input("Selection: ")) if chosen == 1: query = input("Enter new search query: [track/album/video] Darkside Alan Walker: ") break else: exit() print() if query: args.urls = ("search " + query).split() continue if searchtype == 'tracks': media_to_download = [{'id': str(searchresult[searchtype]['items'][chosen]['id']), 'type': 't'}] elif searchtype == 'albums': media_to_download = [{'id': str(searchresult[searchtype]['items'][chosen]['id']), 'type': 'a'}] elif searchtype == 'videos': media_to_download = [{'id': str(searchresult[searchtype]['items'][chosen]['id']), 'type': 'v'}] # elif searchtype == 'playlists': # media_to_download = [{'id': str(searchresult[searchtype]['items'][chosen]['id']), 'type': 'p'}] break else: media_to_download = cli.parse_media_option(args.urls, args.file) print(LOGO) # Loop through media and download if possible cm = 0 for mt in media_to_download: # Is it an acceptable media type? (skip if not) if not mt['type'] in MEDIA_TYPES: print('Unknown media type - ' + mt['type']) continue cm += 1 print('<<< Getting {0} info... >>>'.format(MEDIA_TYPES[mt['type']])) # Create a new TidalApi and pass it to a new MediaDownloader md = MediaDownloader(TidalApi(RSF.load_session(args.account)), preset, Tagger(preset)) # Create a new session generator in case we need to switch sessions session_gen = RSF.get_session() # Get media info def get_tracks(media): media_name = None tracks = [] media_info = None track_info = [] while True: try: if media['type'] == 'f': lines = media['content'].split('\n') for i, l in enumerate(lines): print('Getting info for track {}/{}'.format(i, len(lines)), end='\r') tracks.append(md.api.get_track(l)) print() # Track elif media['type'] == 't': tracks.append(md.api.get_track(media['id'])) # Playlist elif media['type'] == 'p': # Get playlist title to create path playlist = md.api.get_playlist(media['id']) md.opts['path'] += '/' + md._sanitise_name(playlist['title']) # Make sure only tracks are in playlist items playlistItems = md.api.get_playlist_items(media['id'])['items'] for item in playlistItems: if item['type'] == 'track': tracks.append(item['item']) # Album elif media['type'] == 'a': # Get album information media_info = md.api.get_album(media['id']) # Get a list of the tracks from the album tracks = md.api.get_album_tracks(media['id'])['items'] # Video elif media['type'] == 'v': # Get video information tracks.append(md.api.get_video(media['id'])) # Artist else: # Get the name of the artist for display to user media_name = md.api.get_artist(media['id'])['name'] # Collect all of the tracks from all of the artist's albums albums = md.api.get_artist_albums(media['id'])['items'] + md.api.get_artist_albums_ep_singles(media['id'])['items'] eps_info = [] singles_info = [] for album in albums: if 'aggressive_remix_filtering' in preset and preset['aggressive_remix_filtering']: title = album['title'].lower() if 'remix' in title or 'commentary' in title or 'karaoke' in title: print('\tSkipping ' + album['title']) continue # remove sony 360 reality audio albums if there's another (duplicate) album that isn't 360 reality audio if 'skip_360ra' in preset and preset['skip_360ra']: if 'SONY_360RA' in album['audioModes']: is_duplicate = False for a2 in albums: if album['title'] == a2['title'] and album['numberOfTracks'] == a2['numberOfTracks']: is_duplicate = True break if is_duplicate: print('\tSkipping duplicate Sony 360 Reality Audio album - ' + album['title']) continue # Get album information media_info = md.api.get_album(album['id']) # Get a list of the tracks from the album tracks = md.api.get_album_tracks(album['id'])['items'] if 'type' in media_info and str(media_info['type']).lower() == 'single': singles_info.append((tracks, media_info)) else: eps_info.append((tracks, media_info)) if 'skip_singles_when_possible' in preset and preset['skip_singles_when_possible']: # Filter singles that also appear in albums (EPs) def track_in_ep(title): for tracks, _ in eps_info: for t in tracks: if t['title'] == title: return True return False for track_info in singles_info[:]: for t in track_info[0][:]: if track_in_ep(t['title']): print('\tSkipping ' + t['title']) track_info[0].remove(t) if len(track_info[0]) == 0: singles_info.remove(track_info) track_info = eps_info + singles_info if not track_info: track_info = [(tracks, media_info)] return media_name, track_info # Catch region error except TidalError as e: if 'not found. This might be region-locked.' in str(e) and BRUTEFORCE: # Try again with a different session try: session, name = next(session_gen) md.api = TidalApi(session) print('Checking info fetch with session "{}" in region {}'.format(name, session.country_code)) continue # Ran out of sessions except StopIteration as s: print(e) raise s # Skip or halt else: raise(e) try: media_name, track_info = get_tracks(media=mt) except StopIteration: # Let the user know we cannot download this release and skip it print('None of the available accounts were able to get info for release {}. Skipping..'.format(mt['id'])) continue total = sum([len(t[0]) for t in track_info]) # Single if total == 1: print('<<< Downloading single track... >>>') # Playlist or album else: if mt['type'] == 'p': name = md.playlist_from_id(mt['id'])['title'] else: name = track_info[0][1]['title'] print('<<< Downloading {0} "{1}": {2} track(s) in total >>>'.format( MEDIA_TYPES[mt['type']] + (' ' + media_name if media_name else ''), name, total)) if args.resumeon and len(media_to_download) == 1 and mt['type'] == 'p': print('<<< Resuming on track {} >>>'.format(args.resumeon)) args.resumeon -= 1 else: args.resumeon = 0 cur = args.resumeon for tracks, media_info in track_info: for track in tracks[args.resumeon:]: first = True # Actually download the track (finally) while True: try: md.download_media(track, preset, media_info, overwrite=args.overwrite) break # Catch quality error except ValueError as e: print("\t" + str(e)) traceback.print_exc() if args.skip is True: print('Skipping track "{} - {}" due to insufficient quality'.format( track['artist']['name'], track['title'])) break else: print('Halting on track "{} - {}" due to insufficient quality'.format( track['artist']['name'], track['title'])) quit() # Catch file name errors except OSError as e: print(e) print("\tFile name too long or contains apostrophes") file = open('failed_tracks.txt', 'a') file.write(str(track['url']) + "\n") file.close() break # Catch session audio stream privilege error except AssertionError as e: if 'Unable to download track' in str(e) and BRUTEFORCE: # Try again with a different session try: # Reset generator if this is the first attempt if first: session_gen = RSF.get_session() first = False session, name = next(session_gen) md.api = TidalApi(session) print('Attempting audio stream with session "{}" in region {}'.format(name, session.country_code)) continue # Ran out of sessions, skip track except StopIteration: # Let the user know we cannot download this release and skip it print('None of the available accounts were able to download track {}. Skipping..'.format(track['id'])) break elif 'Please use a mobile session' in str(e): print(e) print('Choose one of the following mobile sessions: ') RSF.list_sessions(True) break # Skip else: print(str(e) + '. Skipping..') # Progress of current track cur += 1 print('=== {0}/{1} complete ({2:.0f}% done) ===\n'.format( cur, total, (cur / total) * 100)) # Progress of queue print('> Download queue: {0}/{1} items complete ({2:.0f}% done) <\n'. format(cm, len(media_to_download), (cm / len(media_to_download)) * 100)) print('> All downloads completed. <') # since oauth sessions can change while downloads are happening if the token gets refreshed RSF._save()
def main(): os.chdir(sys.path[0]) # Get args args = cli.get_args() # Check for auth flag / session settings RSF = RedseaSessionFile('./config/sessions.pk') if args.urls[0] == 'auth' and len(args.urls) == 1: print('\nThe "auth" command provides the following methods:') print('\n list: list the currently stored sessions') print(' add: login and store a new session') print(' remove: permanently remove a stored session') print(' default: set a session as default') print(' reauth: reauthenticate with Tidal to get new sessionId') print('\nUsage: redsea.py auth add\n') exit() elif args.urls[0] == 'auth' and len(args.urls) > 1: if args.urls[1] == 'list': RSF.list_sessions() exit() elif args.urls[1] == 'add': if len(args.urls) == 5: RSF.create_session(args.urls[2], args.urls[3], args.urls[4]) else: RSF.new_session() exit() elif args.urls[1] == 'remove': RSF.remove_session() exit() elif args.urls[1] == 'default': RSF.set_default() exit() elif args.urls[1] == 'reauth': RSF.reauth() exit() print(LOGO) # Load config BRUTEFORCE = args.bruteforce or BRUTEFORCEREGION preset = PRESETS[args.preset] # Parse options preset['quality'] = [] preset['quality'].append('HI_RES') if preset['MQA_FLAC_24'] else None preset['quality'].append('LOSSLESS') if preset['FLAC_16'] else None preset['quality'].append('HIGH') if preset['AAC_320'] else None preset['quality'].append('LOW') if preset['AAC_96'] else None media_to_download = cli.parse_media_option(args.urls, args.file) # Loop through media and download if possible cm = 0 for mt in media_to_download: # Is it an acceptable media type? (skip if not) if not mt['type'] in MEDIA_TYPES: print('Unknown media type - ' + mt['type']) continue cm += 1 print('<<< Getting {0} info... >>>'.format(MEDIA_TYPES[mt['type']])) # Create a new TidalApi and pass it to a new MediaDownloader md = MediaDownloader(TidalApi(RSF.load_session(args.account)), preset, Tagger(preset)) # Create a new session generator in case we need to switch sessions session_gen = RSF.get_session() # Get media info def get_tracks(media): media_name = None tracks = [] media_info = None track_info = [] while True: try: if media['type'] == 'f': lines = media['content'].split('\n') for i, l in enumerate(lines): print('Getting info for track {}/{}'.format( i, len(lines)), end='\r') tracks.append(md.api.get_track(l)) print() # Track elif media['type'] == 't': tracks.append(md.api.get_track(media['id'])) # Playlist elif media['type'] == 'p': # Make sure only tracks are in playlist items playlistItems = md.api.get_playlist_items( media['id'])['items'] for item in playlistItems: if item['type'] == 'track': tracks.append(item['item']) # Album elif media['type'] == 'a': # Get album information media_info = md.api.get_album(media['id']) # Get a list of the tracks from the album tracks = md.api.get_album_tracks(media['id'])['items'] # Artist else: # Get the name of the artist for display to user media_name = md.api.get_artist(media['id'])['name'] # Collect all of the tracks from all of the artist's albums albums = md.api.get_artist_albums( media['id'] )['items'] + md.api.get_artist_albums_ep_singles( media['id'])['items'] eps_info = [] singles_info = [] for album in albums: if 'aggressive_remix_filtering' in preset and preset[ 'aggressive_remix_filtering']: title = album['title'].lower() if 'remix' in title or 'commentary' in title or 'karaoke' in title: print('\tSkipping ' + album['title']) continue # remove sony 360 reality audio albums if there's another (duplicate) album that isn't 360 reality audio if 'skip_360ra' in preset and preset['skip_360ra']: if 'SONY_360RA' in album['audioModes']: is_duplicate = False for a2 in albums: if album['title'] == a2[ 'title'] and album[ 'numberOfTracks'] == a2[ 'numberOfTracks']: is_duplicate = True break if is_duplicate: print( '\tSkipping duplicate Sony 360 Reality Audio album - ' + album['title']) continue # Get album information media_info = md.api.get_album(album['id']) # Get a list of the tracks from the album tracks = md.api.get_album_tracks( album['id'])['items'] if 'type' in media_info and str( media_info['type']).lower() == 'single': singles_info.append((tracks, media_info)) else: eps_info.append((tracks, media_info)) if 'skip_singles_when_possible' in preset and preset[ 'skip_singles_when_possible']: # Filter singles that also appear in albums (EPs) def track_in_ep(title): for tracks, _ in eps_info: for t in tracks: if t['title'] == title: return True return False for track_info in singles_info[:]: for t in track_info[0][:]: if track_in_ep(t['title']): print('\tSkipping ' + t['title']) track_info[0].remove(t) if len(track_info[0]) == 0: singles_info.remove(track_info) track_info = eps_info + singles_info if not track_info: track_info = [(tracks, media_info)] return media_name, track_info # Catch region error except TidalError as e: if 'not found. This might be region-locked.' in str( e) and BRUTEFORCE: # Try again with a different session try: session, name = next(session_gen) md.api = TidalApi(session) print( 'Checking info fetch with session "{}" in region {}' .format(name, session.country_code)) continue # Ran out of sessions except StopIteration as s: print(e) raise s # Skip or halt else: raise (e) try: media_name, track_info = get_tracks(media=mt) except StopIteration: # Let the user know we cannot download this release and skip it print( 'None of the available accounts were able to get info for release {}. Skipping..' .format(mt['id'])) continue total = sum([len(t[0]) for t in track_info]) # Single if total == 1: print('<<< Downloading single track... >>>') # Playlist or album else: print('<<< Downloading {0}: {1} track(s) in total >>>'.format( MEDIA_TYPES[mt['type']] + (' ' + media_name if media_name else ''), total)) if args.resumeon and len(media_to_download) == 1 and mt['type'] == 'p': print('<<< Resuming on track {} >>>'.format(args.resumeon)) args.resumeon -= 1 else: args.resumeon = 0 cur = args.resumeon for tracks, media_info in track_info: for track in tracks[args.resumeon:]: first = True # Actually download the track (finally) while True: try: md.download_media(track, preset['quality'], media_info, overwrite=args.overwrite) break # Catch quality error except ValueError as e: print("\t" + str(e)) traceback.print_exc() if args.skip is True: print( 'Skipping track "{} - {}" due to insufficient quality' .format(track['artist']['name'], track['title'])) break else: print( 'Halting on track "{} - {}" due to insufficient quality' .format(track['artist']['name'], track['title'])) quit() # Catch session audio stream privilege error except AssertionError as e: if 'Unable to download track' in str(e) and BRUTEFORCE: # Try again with a different session try: # Reset generator if this is the first attempt if first: session_gen = RSF.get_session() first = False session, name = next(session_gen) md.api = TidalApi(session) print( 'Attempting audio stream with session "{}" in region {}' .format(name, session.country_code)) continue # Ran out of sessions, skip track except StopIteration: # Let the user know we cannot download this release and skip it print( 'None of the available accounts were able to download track {}. Skipping..' .format(track['id'])) break # Skip else: print(str(e) + '. Skipping..') # Progress of current track cur += 1 print('=== {0}/{1} complete ({2:.0f}% done) ===\n'.format( cur, total, (cur / total) * 100)) # Progress of queue print('> Download queue: {0}/{1} items complete ({2:.0f}% done) <\n'. format(cm, len(media_to_download), (cm / len(media_to_download)) * 100)) print('> All downloads completed. <') # since oauth sessions can change while downloads are happening if the token gets refreshed RSF._save()
def main(): # Get args args = cli.get_args() # Check for auth flag / session settings RSF = RedseaSessionFile('./config/sessions.pk') if args.urls[0] == 'auth' and len(args.urls) == 1: print('\nThe "auth" command provides the following methods:') print('\n list: list the currently stored sessions') print(' add: login and store a new session') print(' remove: permanently remove a stored session') print(' default: set a session as default') print(' reauth: reauthenticate with Tidal to get new sessionId') print('\nUsage: redsea.py auth add\n') exit() elif args.urls[0] == 'auth' and len(args.urls) > 1: if args.urls[1] == 'list': RSF.list_sessions() exit() elif args.urls[1] == 'add': RSF.new_session() exit() elif args.urls[1] == 'remove': RSF.remove_session() exit() elif args.urls[1] == 'default': RSF.set_default() exit() elif args.urls[1] == 'reauth': RSF.reauth() exit() print(LOGO) # Load config BRUTEFORCE = args.bruteforce or BRUTEFORCEREGION preset = PRESETS[args.preset] # Parse options preset['quality'] = [] preset['quality'].append('HI_RES') if preset['MQA_FLAC_24'] else None preset['quality'].append('LOSSLESS') if preset['FLAC_16'] else None preset['quality'].append('HIGH') if preset['AAC_320'] else None preset['quality'].append('LOW') if preset['AAC_96'] else None media_to_download = cli.parse_media_option(args.urls) # Loop through media and download if possible cm = 0 for mt in media_to_download: # Is it an acceptable media type? (skip if not) if not mt['type'] in MEDIA_TYPES: print('Unknown media type - ' + mt['type']) continue cm += 1 print('<<< Getting {0} info... >>>'.format(MEDIA_TYPES[mt['type']]), end='\r') # Create a new TidalApi and pass it to a new MediaDownloader md = MediaDownloader(TidalApi(RSF.load_session(args.account)), preset, Tagger(preset)) # Create a new session generator in case we need to switch sessions session_gen = RSF.get_session() # Get media info def get_tracks(media): tracks = [] media_info = None while True: try: # Track if media['type'] == 't': tracks.append(md.api.get_track(media['id'])) # Playlist elif media['type'] == 'p': # Make sure only tracks are in playlist items playlistItems = md.api.get_playlist_items( media['id'])['items'] for item in playlistItems: if item['type'] == 'track': tracks.append(item['item']) # Album else: # Get album information media_info = md.api.get_album(media['id']) # Get a list of the tracks from the album tracks = md.api.get_album_tracks(media['id'])['items'] return tracks, media_info # Catch region error except TidalError as e: if 'not found. This might be region-locked.' in str( e) and BRUTEFORCE: # Try again with a different session try: session, name = next(session_gen) md.api = TidalApi(session) print( 'Checking info fetch with session "{}" in region {}' .format(name, session.country_code)) continue # Ran out of sessions except StopIteration as s: print(e) raise s # Skip or halt else: raise (e) try: tracks, media_info = get_tracks(media=mt) except StopIteration: # Let the user know we cannot download this release and skip it print( 'None of the available accounts were able to get info for release {}. Skipping..' .format(mt['id'])) continue total = len(tracks) # Single if total == 1: print('<<< Downloading single track... >>>') # Playlist or album else: print('<<< Downloading {0}: {1} track(s) in total >>>'.format( MEDIA_TYPES[mt['type']], total)) cur = 0 for track in tracks: first = True # Actually download the track (finally) while True: try: md.download_media(track, preset['quality'], media_info) break # Catch quality error except ValueError as e: print("\t" + str(e)) if args.skip is True: print( 'Skipping track "{} - {}" due to insufficient quality' .format(track['artist']['name'], track['title'])) break else: print( 'Halting on track "{} - {}" due to insufficient quality' .format(track['artist']['name'], track['title'])) quit() # Catch session audio stream privilege error except AssertionError as e: if 'Unable to download track' in str(e) and BRUTEFORCE: # Try again with a different session try: # Reset generator if this is the first attempt if first: session_gen = RSF.get_session() first = False session, name = next(session_gen) md.api = TidalApi(session) print( 'Attempting audio stream with session "{}" in region {}' .format(name, session.country_code)) continue # Ran out of sessions, skip track except StopIteration: # Let the user know we cannot download this release and skip it print( 'None of the available accounts were able to download track {}. Skipping..' .format(track['id'])) break # Skip else: print(str(e) + '. Skipping..') # Progress of current track cur += 1 print('=== {0}/{1} complete ({2:.0f}% done) ===\n'.format( cur, total, (cur / total) * 100)) # Progress of queue print('> Download queue: {0}/{1} items complete ({2:.0f}% done) <\n'. format(cm, len(media_to_download), (cm / len(media_to_download)) * 100)) print('> All downloads completed. <')
def main(): # Get args args = cli.get_args() # Check for auth flag / session settings currentSession = SimpleSessionFile('./config/sessions.pk') deal_with_auth(currentSession, args) print(LOGO) # Load config try: preset = PRESETS[args.preset] except KeyError: print("Unknown preset : " + args.preset) exit(2) # Parse options preset['quality'] = [] preset['quality'].append('HI_RES') if preset['MQA_FLAC_24'] else None preset['quality'].append('LOSSLESS') if preset['FLAC_16'] else None preset['quality'].append('HIGH') if preset['AAC_320'] else None preset['quality'].append('LOW') if preset['AAC_96'] else None media_to_download = cli.parse_media_option(args.urls) # Loop through media and download if possible cm = 0 for mt in media_to_download: # Is it an acceptable media type? (skip if not) if not mt['type'] in MEDIA_TYPES: print('Unknown media type - ' + mt['type']) continue cm += 1 print('<<< Getting {0} info... >>>'.format(MEDIA_TYPES[mt['type']]), end='\r') # Create a new TidalApi and pass it to a new MediaDownloader mediaDownloader = MediaDownloader( TidalApi(currentSession.load_session()), preset, Tagger(preset)) # Get media info try: tracks, media_info = get_tracks(mt, mediaDownloader) except StopIteration: # Let the user know we cannot download this release and skip it print( 'The account is not able to get info for release {}. Skipping..' .format(mt['id'])) continue total = len(tracks) # Single if total == 1: print('<<< Downloading single track... >>>') # Playlist or album or artist else: print('<<< Downloading {0}: {1} track(s) in total >>>'.format( MEDIA_TYPES[mt['type']], total)) cur = 0 for track in tracks: # Actually download the track (finally) while True: try: mediaDownloader.download_media(track, preset['quality'], args.outdir, media_info) break # Catch quality error except ValueError as e: print("\t" + str(e)) if args.skip is True: print( 'Skipping track "{} - {}" due to insufficient quality' .format(track['artist']['name'], track['title'])) break else: print( 'Halting on track "{} - {}" due to insufficient quality' .format(track['artist']['name'], track['title'])) quit() # Catch session audio stream privilege error except AssertionError as e: if args.skip is True: print(str(e) + '. Skipping...') break else: print(str(e) + '. Stopping. Use -s to skip instead') quit() # Progress of current track cur += 1 print('=== {0}/{1} complete ({2:.0f}% done) ===\n'.format( cur, total, (cur / total) * 100)) # Progress of queue print('> Download queue: {0}/{1} items complete ({2:.0f}% done) <\n'. format(cm, len(media_to_download), (cm / len(media_to_download)) * 100)) print('> All downloads completed. <')