298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-13 14:13:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Repair the file operation abnormality and add a log record

Browse Source
This commit is contained in:
hayden 2025-01-24 16:58:40 +08:00
parent 2cf95ed231
commit 9c957977d0
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
app/output_manager.py
View File

@ -1,9 +1,11 @@
from __future__ import annotations
import os
import logging
import folder_paths
import mimetypes
import shutil
import traceback
from aiohttp import web
from concurrent.futures import ThreadPoolExecutor, as_completed
from io import BytesIO
@ -29,8 +31,8 @@ class OutputManager:
def add_routes(self, routes) -> None:
@routes.get("/output{pathname:.*}")
async def get_output_file_or_files(request):
pathname = request.match_info.get("pathname", None)
try:
pathname = request.match_info.get("pathname", None)
filepath = self.get_output_filepath(pathname)
if os.path.isfile(filepath):
@ -51,13 +53,15 @@ class OutputManager:
return web.json_response(files)
return web.Response(status=404)
except Exception as e:
except Exception:
logging.error(f"File '{pathname}' retrieval failed")
logging.error(traceback.format_exc())
return web.Response(status=500)
@routes.delete("/output{pathname:.*}")
async def delete_output_file_or_files(request):
pathname = request.match_info.get("pathname", None)
try:
pathname = request.match_info.get("pathname", None)
filepath = self.get_output_filepath(pathname)
if os.path.isfile(filepath):
@ -66,7 +70,9 @@ class OutputManager:
shutil.rmtree(filepath)
self.rm_cache(filepath)
return web.Response(status=200)
except Exception as e:
except Exception:
logging.error(f"File '{pathname}' deletion failed")
logging.error(traceback.format_exc())
return web.Response(status=500)
def get_output_filepath(self, pathname: str):