Merge pull request #12 from nkchocoai/feature/11-add-date-default

Feature/11 add date default

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:<フォーマット>%` の `<フォーマット>` で指定する識別子は以下の表を参照ください。

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

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