wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-18 21:17:23 +08:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

Merge 40c61a257d into d0f0b15cf5

Browse Source
This commit is contained in:
jiafeimao0p 2026-05-03 20:37:32 +08:00 committed by GitHub
commit 9bd3fc222c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 29 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
server.py
View File

@ -54,6 +54,31 @@ if args.enable_manager:
import comfyui_manager
def _make_content_disposition(filename: str, disposition: str = "inline") -> str:
"""
Generate RFC 8187 compliant Content-Disposition header.
According to RFC 8187, the filename parameter should use the extended notation
with 'filename*=' and 'utf-8'' language tag when the filename contains non-ASCII
characters or needs special encoding.
Examples:
ASCII: inline; filename="test.png"; filename*=UTF-8''test.png
Unicode: inline; filename="__.png"; filename*=UTF-8''%E6%B5%8B%E8%AF%95.png
"""
import urllib.parse
# Create ASCII-safe fallback filename
fallback = "".join(
ch if 0x20 <= ord(ch) < 0x7F and ch not in {'"', '\\'} else "_"
for ch in filename
) or "download"
# Percent-encode the UTF-8 filename
encoded_filename = urllib.parse.quote(filename, safe='')
return f"{disposition}; filename=\"{fallback}\"; filename*=UTF-8''{encoded_filename}"
def _remove_sensitive_from_queue(queue: list) -> list:
"""Remove sensitive data (index 5) from queue item tuples."""
return [item[:5] for item in queue]
@ -559,7 +584,7 @@ class PromptServer():
buffer.seek(0)
return web.Response(body=buffer.read(), content_type=f'image/{image_format}',
headers={"Content-Disposition": f"filename=\"{filename}\""})
headers={"Content-Disposition": _make_content_disposition(filename)})
if 'channel' not in request.rel_url.query:
channel = 'rgba'
@ -579,7 +604,7 @@ class PromptServer():
buffer.seek(0)
return web.Response(body=buffer.read(), content_type='image/png',
headers={"Content-Disposition": f"filename=\"{filename}\""})
headers={"Content-Disposition": _make_content_disposition(filename)})
elif channel == 'a':
with Image.open(file) as img:
@ -596,7 +621,7 @@ class PromptServer():
alpha_buffer.seek(0)
return web.Response(body=alpha_buffer.read(), content_type='image/png',
headers={"Content-Disposition": f"filename=\"{filename}\""})
headers={"Content-Disposition": _make_content_disposition(filename)})
else:
# Use the content type from asset resolution if available,
# otherwise guess from the filename.
@ -613,7 +638,7 @@ class PromptServer():
return web.FileResponse(
file,
headers={
"Content-Disposition": f"filename=\"{filename}\"",
"Content-Disposition": _make_content_disposition(filename),
"Content-Type": content_type
}
)