Major update

Made changes to make nodes more cohesive
2024-07-05 09:04:31 +05:30
parent 37d33ca268
commit e8752c4f71
20 changed files with 441 additions and 543 deletions
@@ -0,0 +1,80 @@
+import os
+
+def save_file(filename, output_dir, content, mode='SaveNew'):
+    os.makedirs(output_dir, exist_ok=True)
+    file_path = os.path.join(output_dir, filename)
+    
+    if mode == 'SaveNew':
+        counter = 0
+        while os.path.exists(file_path):
+            counter += 1
+            file_path = os.path.join(output_dir, f"{os.path.splitext(filename)[0]}_{counter}{os.path.splitext(filename)[1]}")
+    elif mode == 'Merge' and os.path.exists(file_path):
+        with open(file_path, 'a') as file:
+            file.write(content)
+        print(f"Content appended successfully to {file_path}")
+        return
+    elif mode == 'Overwrite' and os.path.exists(file_path):
+        os.remove(file_path)
+    elif mode == 'MergeAndSaveNew' and os.path.exists(file_path):
+        with open(file_path, 'r') as file:
+            existing_content = file.read()
+        content = existing_content + content
+        counter = 0
+        while os.path.exists(file_path):
+            counter += 1
+            file_path = os.path.join(output_dir, f"{os.path.splitext(filename)[0]}_{counter}{os.path.splitext(filename)[1]}")
+    
+    with open(file_path, 'w') as file:
+        file.write(content)
+    print(f"File saved successfully at {file_path}")
+
+class DataSet_TextFilesSave:
+
+    def __init__(self):
+        pass
+
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "TextFileContents": ("STRING",{"forceInput": True}),
+                "TextFileNames": ("STRING",{"forceInput": True}),
+                "destination": ("STRING", {"default": "directory path"}),
+                "save_mode": (['Merge','Overwrite','SaveNew','MergeAndSaveNew'],),
+            },
+        }
+
+    INPUT_IS_LIST = True
+    RETURN_TYPES = ()
+    FUNCTION = "SaveIT"
+    OUTPUT_NODE = True
+
+    CATEGORY = "🔶DATASET🔶"
+
+    def SaveIT(self, TextFileContens, TextFileNames, destination, save_mode):
+        try:
+
+            directory = destination[0]
+            mode = save_mode[0]
+
+            if not os.path.exists(directory):
+                os.makedirs(directory)
+
+            for i in range(0, len(TextFileContens)):
+                text = TextFileContens[i]
+                file_name = TextFileNames[i]
+                save_file(f"{file_name}.txt", directory, text, mode)
+
+        except Exception as e:
+            print(f"Error saving: {e}")
+
+        return ()
+
+N_CLASS_MAPPINGS = {
+    "DataSet_TextFilesSave": DataSet_TextFilesSave,
+}
+
+N_DISPLAY_NAME_MAPPINGS = {
+    "DataSet_TextFilesSave": "DataSet_TextFilesSave",
+}