jinx/ComfyUI
1
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 23:14:49 +08:00
Code Issues Activity

Merge branch 'save-images' of https://github.com/m957ymj75urz/ComfyUI

Browse Source
This commit is contained in:
comfyanonymous
2023-03-20 14:29:45 -04:00
parent 46b674c9c2 d1138e8ba0
commit 0abe001edd
4 changed files with 39 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
server.py
View File

@@ -115,17 +115,24 @@ class PromptServer():
return web.Response(status=400)
@routes.get("/view/{file}")
@routes.get("/view")
async def view_image(request):
if "file" in request.match_info:
if "filename" in request.rel_url.query:
type = request.rel_url.query.get("type", "output")
if type not in ["output", "input", "temp"]:
return web.Response(status=400)
output_dir = os.path.join(os.path.dirname(os.path.realpath(__file__)), type)
file = request.match_info["file"]
if "subfolder" in request.rel_url.query:
full_output_dir = os.path.join(output_dir, request.rel_url.query["subfolder"])
if os.path.commonpath((os.path.realpath(full_output_dir), output_dir)) != output_dir:
return web.Response(status=403)
output_dir = full_output_dir
file = request.rel_url.query["filename"]
file = os.path.basename(file)
file = os.path.join(output_dir, file)
if os.path.isfile(file):
return web.FileResponse(file)