wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-03 10:10:20 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

replace print() to logging.error()

Browse Source
This commit is contained in:
yt-koike 2026-01-30 12:14:28 +09:00
parent 389b3325d1
commit b65c1b1580
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
app/model_manager.py
View File

@ -82,16 +82,16 @@ class ModelFileManager:
@routes.post("/download_model")
async def post_download_model(request):
if not self.is_download_model_enabled():
print("Download Model endpoint is disabled")
logging.error("Download Model endpoint is disabled")
return web.Response(status=403)
json_data = await request.json()
url = json_data.get("url", None)
if url is None:
print("URL is not provided")
logging.error("URL is not provided")
return web.Response(status=401)
save_dir = json_data.get("save_dir", None)
if save_dir not in folder_paths.folder_names_and_paths:
print("Save directory is not valid")
logging.error("Save directory is not valid")
return web.Response(status=401)
filename = json_data.get("filename", url.split("/")[-1])
token = json_data.get("token", None)
@ -113,7 +113,7 @@ class ModelFileManager:
os.rename(tmp_path, save_path)
return web.Response(status=200)
except Exception as e:
print(f"Failed to download model: {e}")
logging.error(f"Failed to download model: {e}")
if os.path.exists(tmp_path):
os.remove(tmp_path)
return web.Response(status=500)