add safety check in case of short extensions

so eg if a two-letter or empty extension is used, `.txt` would break, this `max` call protects that.
parent a71b7b5e
...@@ -575,7 +575,7 @@ def save_image(image, path, basename, seed=None, prompt=None, extension='png', i ...@@ -575,7 +575,7 @@ def save_image(image, path, basename, seed=None, prompt=None, extension='png', i
fullfn_without_extension, extension = os.path.splitext(params.filename) fullfn_without_extension, extension = os.path.splitext(params.filename)
if hasattr(os, 'statvfs'): if hasattr(os, 'statvfs'):
max_name_len = os.statvfs(path).f_namemax max_name_len = os.statvfs(path).f_namemax
fullfn_without_extension = fullfn_without_extension[:max_name_len - len(extension)] fullfn_without_extension = fullfn_without_extension[:max_name_len - max(4, len(extension))]
params.filename = fullfn_without_extension + extension params.filename = fullfn_without_extension + extension
_atomically_save_image(image, fullfn_without_extension, extension) _atomically_save_image(image, fullfn_without_extension, extension)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment