From e8c795c28b5746b95b97bc4b23fabfc0ac9a32cf Mon Sep 17 00:00:00 2001 From: nkchocoai Date: Sat, 30 Mar 2024 12:44:31 +0900 Subject: [PATCH 1/2] Add %date% key --- py/nodes/node.py | 17 ++++++++++++----- 1 file changed, 12 insertions(+), 5 deletions(-) diff --git a/py/nodes/node.py b/py/nodes/node.py index f38acc3..bb9d8b9 100644 --- a/py/nodes/node.py +++ b/py/nodes/node.py @@ -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 From 280daabaa03566892bc3a852d76f1ca7bf026fac Mon Sep 17 00:00:00 2001 From: nkchocoai Date: Sat, 30 Mar 2024 12:54:35 +0900 Subject: [PATCH 2/2] #11 Update README --- README.jp.md | 1 + README.md | 1 + 2 files changed, 2 insertions(+) diff --git a/README.jp.md b/README.jp.md index bf7d501..63e6fcf 100644 --- a/README.jp.md +++ b/README.jp.md @@ -30,6 +30,7 @@ git clone https://github.com/nkchocoai/ComfyUI-SaveImageWithMetaData.git | %nprompt:<文字数n>% | Negative Promptの先頭n文字 | | %model% | Checkpoint名 | | %model:<文字数n>% | Checkpoint名の先頭n文字 | +| %date% | 生成日時(yyyyMMddhhmmss) | | %date:<フォーマット>% | 生成日時 | - `%date:<フォーマット>%` の `<フォーマット>` で指定する識別子は以下の表を参照ください。 diff --git a/README.md b/README.md index 1b3997e..5d97a4f 100644 --- a/README.md +++ b/README.md @@ -32,6 +32,7 @@ git clone https://github.com/nkchocoai/ComfyUI-SaveImageWithMetaData.git | %nprompt:[n]% | First n characters of Negative Prompt | | %model% | Checkpoint name | | %model:[n]% | First n characters of Checkpoint name | +| %date% | Date of generation(yyyyMMddhhmmss) | | %date:[format]% | Date of generation | - See the following table for the identifier specified by `[format]` in `%date:[format]%`.