qnloft
/
ComfyUI
mirror of https://github.com/comfyanonymous/ComfyUI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add a way to overwrite images when uploading.

This commit is contained in:
comfyanonymous 2023-05-11 14:15:13 -04:00
parent dfc74c19d9
commit 8ea165dd1e
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
server.py
View File

@ -127,6 +127,7 @@ class PromptServer():
def image_upload(post, image_save_function=None): def image_upload(post, image_save_function=None):
image = post.get("image") image = post.get("image")
overwrite = post.get("overwrite")
image_upload_type = post.get("type") image_upload_type = post.get("type")
upload_dir = get_dir_by_type(image_upload_type) upload_dir = get_dir_by_type(image_upload_type)
@ -148,11 +149,14 @@ class PromptServer():
split = os.path.splitext(filename) split = os.path.splitext(filename)
filepath = os.path.join(full_output_folder, filename) filepath = os.path.join(full_output_folder, filename)
i = 1 if overwrite is not None and (overwrite == "true" or overwrite == "1"):
while os.path.exists(filepath): pass
filename = f"{split[0]} ({i}){split[1]}" else:
filepath = os.path.join(full_output_folder, filename) i = 1
i += 1 while os.path.exists(filepath):
filename = f"{split[0]} ({i}){split[1]}"
filepath = os.path.join(full_output_folder, filename)
i += 1
if image_save_function is not None: if image_save_function is not None:
image_save_function(image, post, filepath) image_save_function(image, post, filepath)