PK!L@g66vidl/__init__.pyfrom vidl import app def main(): return app.main()PK!==vidl/__main__.pyif __name__ == '__main__': # make the package be able to import itself (vidl) import os,sys,inspect currentdir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe()))) parentdir = os.path.dirname(currentdir) sys.path.insert(0,parentdir) from vidl import app app.main()PK!|ʛ vidl/app.pyimport sys from colorboy import green, cyan, red, magenta, yellow, bright def log(*args, error=False, quiet=False, **named_args): vidl_text = cyan('[vidl]') if error: print(vidl_text, red('error:'), *args, **named_args) quit() elif quiet == False: print(vidl_text, *args, **named_args) script_filename = "vidl" # used to be sys.argv[0], but Poetry changes sys.argv[0] to the full path def vidl_help(): print( '') print(green( 'Usage:')) print( ' '+cyan(script_filename)+' [format] [options] ') print( '') print(green( 'Options:')) print(cyan( ' format ')+'mp3, mp4, wav or m4a. Default mp3.') print(cyan( ' --no-md ')+'Don\'t add metadata to downloaded files.') print(cyan( ' --no-smart-md ')+'Don\'t extract artist and song name from title.') print(cyan( ' -v, --verbose ')+'Display all logs.') print(cyan( ' -h, --help ')+'Display this help message.') print( '') print(green( 'Configuration:')) print( ' '+cyan(script_filename)+' config [new_value]') print( ' ') print(green( 'Available Configs:')) print(cyan( ' download_folder ')+'The folder that vidl downloads to.') print(cyan( ' output_template ')+'youtube-dl output template.') print( '') quit() def main(): if len(sys.argv) <= 1: vidl_help() elif sys.argv[1] == 'config': from vidl import config config.main() else: from vidl import dl dl.main() if __name__ == '__main__': main()PK!(wwwvidl/config.json{ "download_folder": "~/Downloads", "output_template": "%(uploader)s - %(title)s.%(ext)s", "add_metadata": true }PK!{FFvidl/config.pyimport sys, json, os from colorboy import green from vidl.app import vidl_help, log from pprint import pformat def path(*args, **options): DIR = os.path.dirname(os.path.abspath(__file__)) return os.path.join(DIR, *args, **options) configs = json.loads(open(path('config.json')).read()) def save(): file = open(path('config.json'), "w+") file.write(json.dumps(configs, indent=2)) file.close() def set_default_download_folder(): if sys.platform == 'darwin': set('download_folder', '~/Downloads') elif sys.platform == 'win32': import os path = os.path.join(os.getenv('USERPROFILE'), 'Downloads') set('download_folder', path) def load(value): if value == 'download_folder' and configs[value] == None: set_default_download_folder() return configs[value] def set(key, value): if not key in configs: log('Config does not exist:', green(key), error=True) configs[key] = value save() def main(): if len(sys.argv) not in [3, 4]: vidl_help() if len(sys.argv) == 3: key = sys.argv[2] value = load(key) log('Config', key+':', green(pformat(value))) if len(sys.argv) == 4: key = sys.argv[2] value = sys.argv[3] set(key, value) save() log(key, 'was set to:', green(pformat(value)))PK!E:GG vidl/dl.pyimport sys, os import pprint; pprint = pprint.PrettyPrinter(indent=4).pprint import youtube_dl from colorboy import cyan, green from deep_filter import deep_filter from vidl import app, config from vidl.app import log class Dicty(dict): __getattr__ = dict.__getitem__ __setattr__ = dict.__setitem__ def is_int(number): try: int(number) return True except ValueError: return False def main(): options = { 'url': '', 'file_format': 'mp3', 'audio_only': True, 'no_md': False, 'no_smart_md': False, 'verbose': False, 'download_folder': config.load('download_folder'), 'output_template': config.load('output_template'), 'add_metadata': config.load('add_metadata'), } if options['download_folder'] == None: log('download_folder config has not been set. To set it, run '+green(app.script_filename+' config download_folder '), error=True) video_formats = ['mp4'] audio_formats = ['mp3', 'wav', 'm4a'] id3_metadata_formats = ['mp3'] ytdl_output_template = os.path.join(options['download_folder'], options['output_template']) # parse arguments for arg in sys.argv[1:]: if arg in audio_formats: options['audio_only'] = True options['file_format'] = arg elif arg in video_formats: options['audio_only'] = False options['file_format'] = arg elif arg in ['--no-md']: options['no_md'] = True elif arg in ['--no-smart-md']: options['no_smart_md'] = True elif arg in ['-v', '--verbose']: options['verbose'] = True elif arg in ['-h', '--help']: app.vidl_help() elif '.' in arg: options['url'] = arg else: log('Unknown argument:', arg, error=True) if len(sys.argv) <= 1: # no arguments provided app.vidl_help() if options['url'] == '': log('No URL provided', error=True) # get info log('Fetching URL info') ytdl_get_info_options = { 'outtmpl': ytdl_output_template, 'quiet': False if options['verbose'] else True, } with youtube_dl.YoutubeDL(ytdl_get_info_options) as ytdl: try: info_result = ytdl.extract_info(options['url'], download=False) except: quit() if options['verbose']: pprint(info_result) # delete None properties/indexes def callback(value): return value != None cleaned_info_result = deep_filter(info_result.copy(), callback) # restructure if 'entries' in cleaned_info_result: videos = cleaned_info_result['entries'] playlist_info = cleaned_info_result.copy() del playlist_info['entries'] else: videos = [cleaned_info_result] playlist_info = {} # generate ytdl arguments ytdl_args = [] if options['audio_only']: ytdl_args += ['-x'] ytdl_args += ['-f', 'best'] ytdl_args += ['--audio-format', options['file_format']] else: ytdl_args += ['-f', 'bestvideo+bestaudio'] ytdl_args += ['--recode-video', options['file_format']] ytdl_args += ['--audio-quality', '0'] ytdl_args += ['-o', ytdl_output_template] if options['file_format'] in ['mp3', 'm4a', 'mp4']: ytdl_args += ['--embed-thumbnail'] if not options['verbose']: ytdl_args += ['--quiet'] # ytdl_args += [options['url']] video_index = -1 for video in videos: video_index += 0 try: filename = ytdl.prepare_filename(video) except: quit() filename_split = filename.split('.') filename_split[len(filename_split)-1] = options['file_format'] filename = '.'.join(filename_split) log('Downloading') try: youtube_dl.main(ytdl_args+[video['webpage_url']]) except: pass log('Saved as', filename) if options['file_format'] in id3_metadata_formats and not options['no_md']: log('Adding metadata to file') # get artist/title from title parsed_title = {} if 'title' in video and video['title'].count(' - ') == 1: if not options['no_smart_md']: split_title = video['title'].split(' - ') parsed_title['artist'] = split_title[0] parsed_title['title'] = split_title[1] md = Dicty() playlist = True if len(videos) > 1 else False # title if 'track' in video: md.title = video['track'] elif 'title' in parsed_title: md.title = parsed_title['title'] elif 'title' in video: md.title = video['title'] # artist if 'artist' in video: md.artist = video['artist'] if 'artist' in parsed_title: md.artist = parsed_title['artist'] elif 'uploader' in video: md.artist = video['uploader'] if playlist: #album if 'title' in playlist_info: md.album = playlist_info['title'] elif 'playlist_title' in video: md.album = video['playlist_title'] elif 'playlist' in video and type(video['playlist']) == str: md.album = video['playlist'] #album_artist if 'uploader' in playlist_info: md.album_artist = playlist_info['uploader'] elif 'playlist_uploader' in video: md.album_artist = video['playlist_uploader'] # track_number if 'playlist_index' in video: md.track_number = video['playlist_index'] else: md.track_number = video_index+1 # track_count if 'n_entries' in video: md.track_count = video['n_entries'] else: md.track_count = len(videos) # year if 'release_date' in video and is_int(video['release_date'][:4]): md.year = video['release_date'][:4] elif 'publish_date' in video and is_int(video['publish_date'][:4]): md.year = video['publish_date'][:4] elif 'upload_date' in video and is_int(video['upload_date'][:4]): md.year = video['upload_date'][:4] from vidl import md as md_module md_module.add_metadata(filename, md) log('Done') PK!}8ff vidl/md.pyfrom mutagen.id3 import ID3, TIT2, TPE1, TALB, TPE2, TRCK, TCON, TDRC # http://id3.org/id3v2.4.0-frames - COMM comment, USLT lyrics, TCOM composer from pprint import pformat from vidl.app import log def add_metadata(filename, md): tags = ID3(filename) if 'title' in md: tags["TIT2"] = TIT2(encoding=3, text=md.title) if 'artist' in md: tags["TPE1"] = TPE1(encoding=3, text=md.artist) if 'album' in md: tags["TALB"] = TALB(encoding=3, text=md.album) if 'album_artist' in md: tags["TPE2"] = TPE2(encoding=3, text=md.album_artist) if 'track_number' in md: track_number = str(md.track_number) if 'track_count' in md: track_number += '/'+str(md.track_count) tags["TRCK"] = TRCK(encoding=3, text=track_number) if 'genre' in md: tags["TCON"] = TCON(encoding=3, text=md.genre) if 'year' in md: tags["TDRC"] = TDRC(encoding=3, text=md.year) for key, value in md.items(): whitespace = ' ' * (10 - len(key)) log(' '+key+':'+whitespace+pformat(value)) tags.save(filename)PK!H;A%&%vidl-3.0.4.dist-info/entry_points.txtN+I/N.,()*LɱzVy\\PK!HW"TTvidl-3.0.4.dist-info/WHEEL A н#J."jm)Afb~ ڡ5 G7hiޅF4+-3ڦ/̖?XPK!H˻ vidl-3.0.4.dist-info/METADATAXao8_4[o\lm]AEKF"u$e8{CRzp[pf8|3oFow|A+^b~,=Hۺf5a+7׊1YҼeBy[H=dVV-n];#(}k1j F5v2&:PO߿}kZjSIdYzk`n̓ {*jH0~ˌn+VaGaGvb?}><^r^p??\~y%#v`=|^O": 81أ?ML MdyЛ׺C6sGke˲-4R85e.e>gU^̘$ ,?p>-݆Բw%(b½ 6?>:(p/();%o`iگ\}SUb!~‚\(UbX@)tL|дBB(+=4;'a(+P<KNCN< ZP&P*U+ªOQ?>7ݦu@rr7A)랦=nև9:#Bzv}$@AX;Q :0b9 S% 8+]WRjn ʨBC(0#v5R \ .֠EC,L"ϥsnE2~-8]'ɯH`iR╓&숽Pd'vdȈTҵov[Xz4pt7X}6?;v;J5r;NJK0exY ݺ0|'ݒC z ëׯ܅&آg ;ڥ ֛:Y]mءЃ 7| #i1 }v Ytg0@;LHUĞىBDza%9,V;jd41Ҹ\`T ._WC@7-6AQ7ߜZUw|>;oa e읳緾x&);vs=p_nh=HH ɹ X;DкZɺ@Ҥ_Fy<Ԋy+:ގEzҁ=N~?ɬ+v;&a׬u~È'{\>PD (YNx9,%g;0=H3r)|~,3ϗY4֦a BdmٓǷH$X^x rz(+6kc<"@ݻxPK!L@g66vidl/__init__.pyPK!==dvidl/__main__.pyPK!|ʛ vidl/app.pyPK!(wwwvidl/config.jsonPK!{FF? vidl/config.pyPK!E:GG vidl/dl.pyPK!}8ff )vidl/md.pyPK!H;A%&%-vidl-3.0.4.dist-info/entry_points.txtPK!HW"TT.vidl-3.0.4.dist-info/WHEELPK!H˻ .vidl-3.0.4.dist-info/METADATAPK!H҅& \9vidl-3.0.4.dist-info/RECORDPK ;