Slight hack: allow pre-given filenames in threaded_dl.

This commit is contained in:
voussoir 2020-10-25 20:15:11 -07:00
parent 0c481f454f
commit 9e65971829

View file

@ -12,6 +12,10 @@ from voussoirkit import downloady
def clean_url_list(urls):
for url in urls:
if isinstance(url, (tuple, list)):
(url, filename) = url
else:
filename = None
url = url.strip()
if not url:
@ -20,6 +24,9 @@ def clean_url_list(urls):
if url.startswith('#'):
continue
if filename:
yield (url, filename)
else:
yield url
def download_thread(url, filename, *, bytespersecond=None, headers=None, timeout=None):
@ -61,6 +68,9 @@ def threaded_dl(
threads = remove_finished(threads)
time.sleep(0.1)
if isinstance(url, (tuple, list)):
(url, filename) = url
else:
basename = downloady.basename_from_url(url)
extension = os.path.splitext(basename)[1]
filename = filename_format.format(
@ -100,6 +110,8 @@ def threaded_dl_argparse(args):
urls = clipext.resolve(args.url_file)
urls = urls.replace('\r', '').split('\n')
urls = [u.split(' ', 1) if ' ' in u else u for u in urls]
headers = args.headers
if headers is not None:
if len(headers) == 1 and headers[0].startswith('{'):