wangbo/ComfyUI-Manager
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI-Manager.git synced 2025-12-16 10:02:28 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Add a suffix to unfinished files to enable aria2's resumable download function.

Browse Source
This commit is contained in:
Sky Zhang 2025-08-06 00:42:07 +08:00
parent 4980f6ba19
commit 2bcae64cd4
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
glob/manager_downloader.py
View File

@ -81,6 +81,7 @@ def aria2_download_url(model_url: str, model_dir: str, filename: str):
import time
download_dir = model_dir
filename += '.downloading' # aria2 uses .downloading suffix for incomplete files
download = aria2_find_task(download_dir, filename)
if download is None:
@ -101,6 +102,17 @@ def aria2_download_url(model_url: str, model_dir: str, filename: str):
progress_bar.update(download.completed_length - progress_bar.n)
time.sleep(1)
download.update()
if download.is_complete and download.completed_length == download.total_length:
file_path = os.path.join(download_dir, filename)
final_path = file_path[:-len('.downloading')] # remove the .downloading suffix
if os.path.exists(file_path):
os.rename(file_path, final_path)
logging.info(f"Download completed: {final_path}")
else:
logging.error(f"Download directory {file_path} does not exist after download completion.")
else:
logging.error(f"Download failed or incomplete: {download.status} for {model_url}")
def download_url_with_agent(url, save_path):