mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-07-05 14:10:45 +08:00
Remove sending url info over websockets for model downloads.
This commit is contained in:
parent
f7627e6b59
commit
84ec4755c6
@ -1210,7 +1210,12 @@ class PromptServer():
|
||||
try:
|
||||
view = DOWNLOAD_MANAGER.status_sync(download_id)
|
||||
if view is not None:
|
||||
self.send_sync("download_progress", view)
|
||||
# Drop the url field before broadcasting: the redacted URL
|
||||
# (scheme + host + path) should not leak to every connected
|
||||
# websocket client. download_id / model_id are sufficient to
|
||||
# correlate progress on the frontend.
|
||||
broadcast = {k: v for k, v in view.items() if k != "url"}
|
||||
self.send_sync("download_progress", broadcast)
|
||||
except Exception:
|
||||
logging.debug("download progress notify failed", exc_info=True)
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user