Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
ComfyUI
Commits
8ea165dd
Commit
8ea165dd
authored
May 11, 2023
by
comfyanonymous
Browse files
Add a way to overwrite images when uploading.
parent
dfc74c19
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
9 additions
and
5 deletions
+9
-5
server.py
server.py
+9
-5
No files found.
server.py
View file @
8ea165dd
...
...
@@ -127,6 +127,7 @@ class PromptServer():
def
image_upload
(
post
,
image_save_function
=
None
):
image
=
post
.
get
(
"image"
)
overwrite
=
post
.
get
(
"overwrite"
)
image_upload_type
=
post
.
get
(
"type"
)
upload_dir
=
get_dir_by_type
(
image_upload_type
)
...
...
@@ -148,11 +149,14 @@ class PromptServer():
split
=
os
.
path
.
splitext
(
filename
)
filepath
=
os
.
path
.
join
(
full_output_folder
,
filename
)
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
overwrite
is
not
None
and
(
overwrite
==
"true"
or
overwrite
==
"1"
):
pass
else
:
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
:
image_save_function
(
image
,
post
,
filepath
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment