def run_one(url):
import threading
flush_info('done queuing ' + url)
cmd = [
'youtube-dl',
# url, '--output', rf'{DOWNLOAD_TO}/%(release_date)s.%(title)s.%(ext)s',
url,
'--output',
rf'{DOWNLOAD_TO}/%(title)s.%(id)s.%(ext)s',
# meta
'--write-info-json',
# '--write-description',
# '--write-annotations',
# sub
'--all-subs',
'--embed-subs'
]
if YDL_PROXY:
cmd += ['--proxy', YDL_PROXY]
subprocess.call(cmd)
flush_info('done with' + cmd)
def persist(s):
s = s.strip()
global g_urls
g_urls.add(s)
flush_info('added ' + s + ('[new]' if s in g_urls else ''))
flush_info('Historical Items:')
for i, u in enumerate(g_urls):
flush_info(f' {i+1}. {u}')
g_cache.write_text('\n'.join(g_urls), encoding='utf-8')
def on_resume(executor):
for name, url in list_part_files(DOWNLOAD_TO).items():
flush_info(f'resuming {name}')
if url:
executor.submit(run_one, url)
def logged_flush():
stdout.flush()
flush_info(str(datetime.now()))
def print_dict(d):
flush_info('\n'.join(f'- {k}, {v}' for k, v in d.items()))
def on_ls_complete(executor):
print_dict(list_downloaded_files(DOWNLOAD_TO))
def print_dict(d):
flush_info('\n'.join(f'- {k}, {v}' for k, v in d.items()))
def print_commands():
return '\n'.join(f'- {k}, {v[0]}' for k, v in commands().items())
def commands():
return {
'resume': ("Resume downloading", on_resume),
'ls-part': ("List partially downloaded files", on_ls_part),
'ls-complete': ("List fully downloaded files", on_ls_complete)
}
if __name__ == "__main__":
enable_logging.init()
flush_info(
f"\nProxy: {YDL_PROXY}, Download folder: {DOWNLOAD_TO}.\nCommands:\n{print_commands()}\n"
)
main()