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!Nga'' vidl/app.pyfrom colorboy import green, cyan, red from pathlib import Path package_name = 'vidl' # used for getting config location and package version package_author = 'Kasper Henningsen' # used for getting config location def log(*args, error=False, quiet=False, callback=None, quit=True, **named_args): vidl_text = cyan('[vidl]') if error: print(vidl_text, red('Error:'), *args, **named_args) 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( 'Download Usage:')) print( ' '+cyan(script_filename)+' [format] [options] ') print( '') print(green( 'Download 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( ' --no-dl ')+'Don\'t download anything. Usually used with -v') print(cyan( ' -v, --verbose ')+'Display all logs.') print( '') print(green( 'Global Options:')) print(cyan( ' --version ')+'Display vidl version. "vidl -v" also works.') print(cyan( ' -h, --help ')+'Display this help message.') print(cyan( ' --config-path ')+'Display the location of vidl\'s configuration file.') print( '') quit() def main(): import sys if len(sys.argv) <= 1 or '--help' in sys.argv or '-h' in sys.argv: vidl_help() elif '--version' in sys.argv or sys.argv[1:] == ['-v']: import vidl.version log("Version", vidl.version.get_package_version()) elif '--config-path' in sys.argv: import vidl.config log("Config path:", vidl.config.config_path) else: from vidl import dl dl.main() if __name__ == '__main__': main() PK!1سvidl/config.pyimport sys, json, os, appdirs from colorboy import green from vidl.app import log, package_name, package_author user_data_dir = appdirs.user_data_dir(package_name, package_author) config_path = os.path.join(user_data_dir, 'config.json') user_md_parser_path = os.path.join(user_data_dir, 'user_md_parser.py') default_user_md_parser_path = os.path.join(os.path.dirname(__file__), 'default_user_md_parser.py') def save_file(path, content, json=False): try: file = open(path, 'w+') except FileNotFoundError: os.makedirs(user_data_dir) file = open(path, 'w+') if json: file.write(json.dumps(content, indent=2)) else: file.write(content) file.close() def get_default_download_folder(): if sys.platform == 'darwin': return '~/Downloads' elif sys.platform == 'win32': import os path = os.path.join(os.getenv('USERPROFILE'), 'Downloads') return path else: return None default_config = { 'download_folder': get_default_download_folder(), 'output_template': '%(uploader)s - %(title)s.%(ext)s', } if not os.path.isfile(config_path): save_file(config_path, default_config) if not os.path.isfile(user_md_parser_path): default_user_md_parser = open(default_user_md_parser_path).read() save_file(user_md_parser_path, default_user_md_parser) configs = json.loads(open(config_path).read()) from importlib.machinery import SourceFileLoader user_md_parser = SourceFileLoader('user_md_parser', user_md_parser_path).load_module().user_md_parser def get_config(key): if key not in configs: log('Config does not exist:', green(key), error=True) quit() return configs[key] PK!V__vidl/default_user_md_parser.pydef user_md_parser(smart_md, md, video_info, playlist_info): # smart_md: # Metadata object created by vidl. Metadata objects can have these properties: # - title # - artist # - album # - album_artist # - track_number # - track_count # - year # - comment # - lyrics # - composer # md: # Same as "md", except this will never include smart metadata (artist/title # parsed from title). # video_info: # Object containing metadata from youtube-dl about the current video. # If there is no info, it equals {}. # playlist_info: # An object containing playlist metadata from youtube-dl. # If the URL isn't a playlist, or there is no playlist info, it equals {}. # callback: Callback function. Takes a metadata object as argument. # example: # This example cheks if vidl has detected title and artist metadata. If the # title includes "[NCS Release]", it will set the comment metadata to "NCS". # if 'title' in md and 'artist' in md: # if '[NCS Release]' in md['title']: # md['comment'] = 'NCS' return md PK!A` 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, 'no_dl': False, 'verbose': False, 'download_folder': config.get_config('download_folder'), 'output_template': config.get_config('output_template'), } if options['download_folder'] == None: log('download_folder config has not been set. Add a download folder to the vidl config file.', error=True) log("Config path:", config.config_path) quit() 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 ['--no-dl']: options['no_dl'] = True elif arg in ['-v', '--verbose']: options['verbose'] = True elif '.' in arg: options['url'] = arg else: log('Unknown argument:', arg, error=True) quit() if options['url'] == '': log('No URL provided', error=True) quit() # 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) if options['verbose']: log(green('youtube-dl command:'), 'youtube-dl', ' '.join(ytdl_args+[video['webpage_url']])) if options['no_dl']: continue 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 not options['no_smart_md']: if 'title' in video and video['title'].count(' - ') == 1: 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 'title' in parsed_title: smart_title = True if 'title' in video: md.title = video['title'] elif 'track' in video: md.title = video['track'] # artist if 'artist' in parsed_title: smart_artist = True if 'uploader' in video: md.artist = video['uploader'] elif 'artist' in video: md.artist = video['artist'] 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] dumb_md = md if smart_title: md['title'] = parsed_title['title'] if smart_artist: md['artist'] = parsed_title['artist'] md = config.user_md_parser(md, dumb_md, video, playlist_info) from vidl import md as md_module md_module.add_metadata(filename, md) log('Done') PK!~H vidl/md.pyfrom mutagen.id3 import ID3, TIT2, TPE1, TALB, TPE2, TRCK, TCON, TDRC, COMM, USLT, TCOM 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) if 'comment' in md: tags["COMM"] = COMM(encoding=3, text=md.comment) if 'lyrics' in md: tags["USLT"] = USLT(encoding=3, text=md.lyrics) if 'composer' in md: tags["TCOM"] = TCOM(encoding=3, text=md.composer) for key, value in md.items(): whitespace = ' ' * (10 - len(key)) log(' '+key+':'+whitespace+pformat(value)) tags.save(filename) PK!vidl/version.pyfrom vidl.app import package_name def get_package_version(): import pkg_resources my_version = pkg_resources.get_distribution(package_name).version return my_version PK!H;A%&%vidl-3.3.0.dist-info/entry_points.txtN+I/N.,()*LɱzVy\\PK!HW"TTvidl-3.3.0.dist-info/WHEEL A н#J."jm)Afb~ ڡ5 G7hiޅF4+-3ڦ/̖?XPK!H(>x vidl-3.3.0.dist-info/METADATAYms6_*=#QyqTǭk7'nⱝt<! 5 (Ess]Ӧbw/\z9zj,Dt^$gqrՔ7qKkպ[۵)icŴхkbcLJ_R*&K+7iV*693e޾}ͯWju&aېwYrz"^GҮezaAҙ-K{kF8z,}:<<'tNϵ.jY,Ĺ4I|zgW|+crYU;z$=>MߗB$HoӃ?M8}vJU9GQveᄉ=-VS; "ӓĞ>~ ߒ RXM&n)%A(%\90*RQbbļfV $47-g^Rz=rr')Ihz6.qy B\ֺ(sQAJsۘ<K%Qg²7$*L`~.uoCs$xCk^YvXLU7w_٭0YYl/5L |`9*qVlSޑ5.nˏJdMf^T颔cOPMRmꔒ8c$cOʟMh:ւv[Z~QkӍ^ivgKQX{Do$ɲ[ w. aM(e^K߄Ll/@d%WTT@Tl6Vl>RSywT84%) *"8(x?.֥AS;kmGXS"ԡߩbXQ{(h;3Cɸ|M  1AӖ1ac9 ѡXELQД1DJ-SyShsв!GZ8O)RL5n@jm %1zO(T@ ;26Sd"nSmSXq!U XjN ͔xlfW4oղxo0p;gXx2( Vf; UWk?l x\R@i@v/FL$*C ڳЀޢ.SlQ2E%Q90'q kT<w"hD{y]yg5,[*8^R4*BϷX!wgN.&n/~ vqǟvŦ{\6FFaYoW*;$-X@ fBp/4(o6~M{ۄlGė>T_ ~plEi琇¶.V{\mP0%f=5H~AL& ՝6mfo7`}S=Ayg@(&@bj8LTʦnf<$~Vz W=U=)v:KWeQsY-!"L }R}yI> 6]cUfRmFolu2ĎpӉ'$6>F/ХkkRuZecʛOb{-(iqՉw ͒H}""BV4m0'{/q>&aQb'* P'\)@$&^A r|>N˞[ks Z-5u)&vD54Pt6Q{qح1Qo'v!F3tK0Q7G\3^>鏈%MBAQ.b%Y!ٵ=H :xscqT9=xb )ޛ۷? ܡ>? F/MؠhƜ_a{n#_Z30`aY '${Ze;RUj<ȏXk/@;_#DOϋz4ʲ$PK!HIJ2;_vidl-3.3.0.dist-info/RECORDmr@@} - aͫ!**:ۓxGTChQOTK~Cf%3k 45!c˺o H}*tи喭m 9'OBQX |;A]DTYX>Za٨pݠ n#wSKF|V%$}aOeb -4 ]w}XMLfh+f8 Emܠ2Bu6_a.--)W'ߴ-)rfbLa }3r@z P=IH|\oč8VlzJPx ~8vidl-3.3.0.dist-info/METADATAPK!HIJ2;_Dvidl-3.3.0.dist-info/RECORDPK G