nodes/nkchocoai-ComfyUI-SaveImageWithMetaData
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add %date% key

Browse Source
This commit is contained in:
nkchocoai
2024-03-30 12:44:31 +09:00
parent 7914a045c9
commit e8c795c28b
1 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
py/nodes/node.py
+12 -5
View File
@@ -175,8 +175,7 @@ class SaveImageWithMetaData(BaseNode):
length = int(parts[1])
model = model[:length]
filename = filename.replace(segment, model)
elif key == "date" and len(parts) >= 2:
date_format = parts[1]
elif key == "date":
now = datetime.now()
date_table = {
"yyyy": now.year,
@@ -186,7 +185,15 @@ class SaveImageWithMetaData(BaseNode):
"mm": now.minute,
"ss": now.second,
}
for k, v in date_table.items():
date_format = date_format.replace(k, str(v).zfill(len(k)))
filename = filename.replace(segment, date_format)
if len(parts) >= 2:
date_format = parts[1]
for k, v in date_table.items():
date_format = date_format.replace(k, str(v).zfill(len(k)))
filename = filename.replace(segment, date_format)
else:
date_format = "yyyyMMddhhmmss"
for k, v in date_table.items():
date_format = date_format.replace(k, str(v).zfill(len(k)))
filename = filename.replace(segment, date_format)
return filename