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]%`. 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