#!/usr/bin/env python # -*- coding: UTF-8 -*- from core import metadata from core import misc from bs4 import BeautifulSoup from titlecase import titlecase from slugify import slugify import spotipy import spotipy.oauth2 as oauth2 import pafy import sys import os import subprocess try: import urllib2 except ImportError: import urllib.request as urllib2 def generate_songname(raw_song): if misc.is_spotify(raw_song): tags = generate_metadata(raw_song) raw_song = tags['artists'][0]['name'] + ' - ' + tags['name'] return misc.fix_encoding(raw_song) def generate_metadata(raw_song): if misc.is_spotify(raw_song): meta_tags = spotify.track(raw_song) else: meta_tags = spotify.search(raw_song, limit=1)['tracks']['items'][0] artist_id = spotify.artist(meta_tags['artists'][0]['id']) try: meta_tags[u'genre'] = titlecase(artist_id['genres'][0]) except IndexError: meta_tags[u'genre'] = None meta_tags[u'release_date'] = spotify.album(meta_tags['album']['id'])['release_date'] meta_tags[u'copyright'] = spotify.album(meta_tags['album']['id'])['copyrights'][0]['text'] meta_tags[u'publisher'] = spotify.album(meta_tags['album']['id'])['label'] meta_tags[u'total_tracks'] = spotify.album(meta_tags['album']['id'])['tracks']['total'] #import pprint #pprint.pprint(meta_tags) #pprint.pprint(spotify.album(meta_tags['album']['id'])) return meta_tags def generate_YouTube_URL(raw_song): song = generate_songname(raw_song) searchURL = misc.generate_search_URL(song) item = urllib2.urlopen(searchURL).read() items_parse = BeautifulSoup(item, "html.parser") check = 1 if args.manual: links = [] print(song) print('') print('0. Skip downloading this song') for x in items_parse.find_all('h3', {'class': 'yt-lockup-title'}): if not x.find('channel') == -1 or not x.find('googleads') == -1: print(str(check) + '. ' + x.get_text()) links.append(x.find('a')['href']) check += 1 print('') result = misc.input_link(links) if result is None: return None else: result = items_parse.find_all( attrs={'class': 'yt-uix-tile-link'})[0]['href'] while not result.find('channel') == - \ 1 or not result.find('googleads') == -1: result = items_parse.find_all( attrs={'class': 'yt-uix-tile-link'})[check]['href'] check += 1 full_link = "youtube.com" + result return full_link def go_pafy(raw_song): trackURL = generate_YouTube_URL(raw_song) if trackURL is None: return None else: return pafy.new(trackURL) def get_YouTube_title(content, number): title = content.title if number is None: return title else: return str(number) + '. ' + title def feed_tracks(file, tracks): with open(file, 'a') as fout: for item in tracks['items']: track = item['track'] try: fout.write(track['external_urls']['spotify'] + '\n') except KeyError: pass def feed_playlist(username): playlists = spotify.user_playlists(username) links = [] check = 1 for playlist in playlists['items']: print(str(check) + '. ' + misc.fix_encoding(playlist['name']) + ' (' + str(playlist['tracks']['total']) + ' tracks)') links.append(playlist) check += 1 print('') playlist = misc.input_link(links) results = spotify.user_playlist(playlist['owner']['id'], playlist['id'], fields="tracks,next") print('') file = slugify(playlist['name'], ok='-_()[]{}') + '.txt' print('Feeding ' + str(playlist['tracks']['total']) + ' tracks to ' + file) tracks = results['tracks'] feed_tracks(file, tracks) while tracks['next']: tracks = spotify.next(tracks) feed_tracks(file, tracks) def download_song(content): music_file = misc.generate_filename(content.title) if args.input_ext == '.webm': link = content.getbestaudio(preftype='webm') if link is not None: link.download(filepath='Music/' + music_file + args.input_ext) else: link = content.getbestaudio(preftype='m4a') if link is not None: link.download(filepath='Music/' + music_file + args.input_ext) def convert_song(music_file): if not args.input_ext == args.output_ext: print('Converting ' + music_file + args.input_ext + ' to ' + args.output_ext[1:]) if args.ffmpeg: convert_with_FFmpeg(music_file) else: convert_with_libav(music_file) def convert_with_libav(music_file): if os.name == 'nt': avconv_path = 'Scripts\\avconv.exe' else: avconv_path = 'avconv' if args.verbose: level = 'debug' else: level = '0' #print([avconv_path, # '-loglevel', level, # '-i', 'Music/' + music_file + args.input_ext, # '-ab', '192k', # 'Music/' + music_file + args.output_ext]) subprocess.call([avconv_path, '-loglevel', level, '-i', 'Music/' + music_file + args.input_ext, '-ab', '192k', 'Music/' + music_file + args.output_ext]) os.remove('Music/' + music_file + args.input_ext) def convert_with_FFmpeg(music_file): # What are the differences and similarities between ffmpeg, libav, and avconv? # https://stackoverflow.com/questions/9477115 # ffmeg encoders high to lower quality # libopus > libvorbis >= libfdk_aac > aac > libmp3lame # libfdk_aac due to copyrights needs to be compiled by end user # on MacOS brew install ffmpeg --with-fdk-aac will do just that. Other OS? # https://trac.ffmpeg.org/wiki/Encode/AAC # if os.name == "nt": ffmpeg_pre = 'Scripts//ffmpeg.exe ' else: ffmpeg_pre = 'ffmpeg ' ffmpeg_pre += '-y ' if not args.verbose: ffmpeg_pre += '-hide_banner -nostats -v panic ' if args.input_ext == '.m4a': if args.output_ext == '.mp3': ffmpeg_params = ' -codec:v copy -codec:a libmp3lame -q:a 2 ' elif output_ext == '.webm': ffmpeg_params = ' -c:a libopus -vbr on -b:a 192k -vn ' else: return elif args.input_ext == '.webm': if args.output_ext == '.mp3': ffmpeg_params = ' -ab 192k -ar 44100 -vn ' elif args.output_ext == '.m4a': ffmpeg_params = ' -cutoff 20000 -c:a libfdk_aac -b:a 256k -vn ' else: return else: print('Unknown formats. Unable to convert.', args.input_ext, args.output_ext) return #command = (ffmpeg_pre + # '-i "Music/' + music_file + args.input_ext + '"' + # ffmpeg_params + # '"Music/' + music_file + args.output_ext + '"').split(' ') commandos = (ffmpeg_pre + '-i "Music/' + music_file + args.input_ext + '" ' + ffmpeg_params + '"Music/' + music_file + args.output_ext + '" ') #print(command) #print(commandos) os.system(commandos) #subprocess.call(command) os.remove('Music/' + music_file + args.input_ext) def check_exists(music_file, raw_song, islist): files = os.listdir("Music") for file in files: if file.endswith(".temp"): os.remove("Music/" + file) continue if file.startswith(misc.generate_filename(music_file)): already_tagged = metadata.compare(file, generate_metadata(raw_song)) if misc.is_spotify(raw_song) and not already_tagged: os.remove("Music/" + file) return False if islist: return True else: prompt = raw_input('Song with same name has already been downloaded. Re-download? (y/n): ').lower() if prompt == "y": os.remove("Music/" + file) return False else: return True def grab_list(file): with open(file, 'r') as listed: lines = (listed.read()).splitlines() # Ignore blank lines in file (if any) try: lines.remove('') except ValueError: pass print('Total songs in list = ' + str(len(lines)) + ' songs') print('') # Count the number of song being downloaded number = 1 for raw_song in lines: try: grab_single(raw_song, number=number) except spotipy.oauth2.SpotifyOauthError: token = misc.generate_token() global spotify spotify = spotipy.Spotify(auth=token) grab_single(raw_song, number=number) except (urllib2.URLError, TypeError, IOError): lines.append(raw_song) misc.trim_song(file) with open(file, 'a') as myfile: myfile.write(raw_song) print('Failed to download song. Will retry after other songs.') continue except KeyboardInterrupt: misc.grace_quit() finally: print('') misc.trim_song(file) number += 1 # Logic behind preparing the song to download to finishing meta-tags def grab_single(raw_song, number=None): if number: islist = True else: islist = False content = go_pafy(raw_song) if content is None: return print(get_YouTube_title(content, number)) music_file = misc.generate_filename(content.title) if not check_exists(music_file, raw_song, islist=islist): download_song(content) print('') convert_song(music_file) meta_tags = generate_metadata(raw_song) if not args.no_metadata: metadata.embed(music_file, meta_tags, args.output_ext) if __name__ == '__main__': # Python 3 compatibility if sys.version_info > (3, 0): raw_input = input os.chdir(sys.path[0]) if not os.path.exists("Music"): os.makedirs("Music") token = misc.generate_token() spotify = spotipy.Spotify(auth=token) # Set up arguments args = misc.get_arguments() if args.song: grab_single(raw_song=args.song) elif args.list: grab_list(file=args.list) elif args.username: feed_playlist(username=args.username)