unshackle-dl/unshackle
1
0
Fork 0
forked from kenzuya/unshackle
Code Pull Requests 1 Activity

fix(downloader): restore requests progress for single-url downloads

Browse Source
This commit is contained in:
Andy
2026-02-06 23:34:15 -07:00
parent ace89760e7
commit 03c309303c
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
unshackle/core/downloaders/requests.py
View File

@@ -260,11 +260,18 @@ def requests(
}, },
) )
# If we're downloading more than one URL, treat them as "segments" for progress purposes.
# For single-URL downloads we want per-chunk progress (and the inner `download()` will yield
# a chunk-based `total`), so don't set a segment total of 1 here.
segmented_batch = len(urls) > 1
if segmented_batch:
yield dict(total=len(urls)) yield dict(total=len(urls))
try: try:
with ThreadPoolExecutor(max_workers=max_workers) as pool: with ThreadPoolExecutor(max_workers=max_workers) as pool:
for future in as_completed(pool.submit(download, session=session, segmented=True, **url) for url in urls): for future in as_completed(
pool.submit(download, session=session, segmented=segmented_batch, **url) for url in urls
):
try: try:
yield from future.result() yield from future.result()
except KeyboardInterrupt: except KeyboardInterrupt: