qnloft
/
ComfyUI
mirror of https://github.com/comfyanonymous/ComfyUI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Revert download to .tmp in frontend_management (#5369)

This commit is contained in:
Chenlei Hu 2024-10-25 19:26:13 -04:00 committed by GitHub
parent 52810907e2
commit ce759b7db6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
app/frontend_management.py
View File

@ -168,20 +168,16 @@ class FrontendManager:
Path(cls.CUSTOM_FRONTENDS_ROOT) / provider.folder_name / semantic_version Path(cls.CUSTOM_FRONTENDS_ROOT) / provider.folder_name / semantic_version
) )
if not os.path.exists(web_root): if not os.path.exists(web_root):
# Use tmp path until complete to avoid path exists check passing from interrupted downloads
tmp_path = web_root + ".tmp"
try: try:
os.makedirs(tmp_path, exist_ok=True) os.makedirs(web_root, exist_ok=True)
logging.info( logging.info(
"Downloading frontend(%s) version(%s) to (%s)", "Downloading frontend(%s) version(%s) to (%s)",
provider.folder_name, provider.folder_name,
semantic_version, semantic_version,
tmp_path, web_root,
) )
logging.debug(release) logging.debug(release)
download_release_asset_zip(release, destination_path=tmp_path) download_release_asset_zip(release, destination_path=web_root)
if os.listdir(tmp_path):
os.rename(tmp_path, web_root)
finally: finally:
# Clean up the directory if it is empty, i.e. the download failed # Clean up the directory if it is empty, i.e. the download failed
if not os.listdir(web_root): if not os.listdir(web_root):