wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-05 11:10:16 +08:00
Code Issues Actions 14 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") @routes.post("/download_model")
async def post_download_model(request): async def post_download_model(request):
if not self.is_download_model_enabled(): 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) return web.Response(status=403)
json_data = await request.json() json_data = await request.json()
url = json_data.get("url", None) url = json_data.get("url", None)
if url is None: if url is None:
print("URL is not provided") logging.error("URL is not provided")
return web.Response(status=401) return web.Response(status=401)
save_dir = json_data.get("save_dir", None) save_dir = json_data.get("save_dir", None)
if save_dir not in folder_paths.folder_names_and_paths: 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) return web.Response(status=401)
filename = json_data.get("filename", url.split("/")[-1]) filename = json_data.get("filename", url.split("/")[-1])
token = json_data.get("token", None) token = json_data.get("token", None)
@ -113,7 +113,7 @@ class ModelFileManager:
os.rename(tmp_path, save_path) os.rename(tmp_path, save_path)
return web.Response(status=200) return web.Response(status=200)
except Exception as e: 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): if os.path.exists(tmp_path):
os.remove(tmp_path) os.remove(tmp_path)
return web.Response(status=500) return web.Response(status=500)