Hawk Lee
e21a5c9eba
The eff537d commit changed all files from 755 to 644.
Restores original executable permissions.
92 lines
3.1 KiB
Python
Executable File
92 lines
3.1 KiB
Python
Executable File
import os
|
|
import time
|
|
import folder_paths
|
|
|
|
class AIIA_TextDebugSplicer:
|
|
NODE_NAME = "Text Debug Splicer (AIIA)"
|
|
CATEGORY = "AIIA/Debug"
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {
|
|
"required": {
|
|
"title_1": ("STRING", {"default": "Text 1", "multiline": False}),
|
|
"title_2": ("STRING", {"default": "Text 2", "multiline": False}),
|
|
"title_3": ("STRING", {"default": "Text 3", "multiline": False}),
|
|
"separator": (["New Line (\\n)", "Double New Line (\\n\\n)", "Dash Line (---)", "Hash Line (###)", "Star Line (***)"], {"default": "Dash Line (---)"}),
|
|
"save_prefix": ("STRING", {"default": "debug_spliced"}),
|
|
"save_file": ("BOOLEAN", {"default": True}),
|
|
"return_text": ("BOOLEAN", {"default": True}),
|
|
},
|
|
"optional": {
|
|
"text_1": ("STRING", {"forceInput": True}),
|
|
"text_2": ("STRING", {"forceInput": True}),
|
|
"text_3": ("STRING", {"forceInput": True}),
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("STRING",)
|
|
RETURN_NAMES = ("combined_text",)
|
|
FUNCTION = "splice_text"
|
|
OUTPUT_NODE = True
|
|
|
|
def splice_text(self, title_1, title_2, title_3, separator, save_prefix, save_file, return_text, text_1=None, text_2=None, text_3=None):
|
|
|
|
# 1. Parsing Separator
|
|
sep_map = {
|
|
"New Line (\\n)": "\n",
|
|
"Double New Line (\\n\\n)": "\n\n",
|
|
"Dash Line (---)": "\n---\n",
|
|
"Hash Line (###)": "\n###\n",
|
|
"Star Line (***)": "\n***\n"
|
|
}
|
|
|
|
actual_sep = sep_map.get(separator, "\n")
|
|
|
|
parts = []
|
|
|
|
# 2. Building Parts
|
|
if text_1:
|
|
if title_1.strip():
|
|
parts.append(f"[{title_1}]\n{text_1}")
|
|
else:
|
|
parts.append(text_1)
|
|
|
|
if text_2:
|
|
if title_2.strip():
|
|
parts.append(f"[{title_2}]\n{text_2}")
|
|
else:
|
|
parts.append(text_2)
|
|
|
|
if text_3:
|
|
if title_3.strip():
|
|
parts.append(f"[{title_3}]\n{text_3}")
|
|
else:
|
|
parts.append(text_3)
|
|
|
|
combined_text = actual_sep.join(parts)
|
|
|
|
# 3. Saving to File
|
|
if save_file and combined_text.strip():
|
|
output_dir = folder_paths.get_output_directory()
|
|
timestamp = time.strftime("%Y%m%d_%H%M%S")
|
|
filename = f"{save_prefix}_{timestamp}.txt"
|
|
full_path = os.path.join(output_dir, filename)
|
|
|
|
try:
|
|
with open(full_path, "w", encoding="utf-8") as f:
|
|
f.write(combined_text)
|
|
print(f"[AIIA Debug] Saved text to {full_path}")
|
|
except Exception as e:
|
|
print(f"[AIIA Debug] Failed to save text: {e}")
|
|
|
|
return (combined_text if return_text else "",)
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"AIIA_TextDebugSplicer": AIIA_TextDebugSplicer
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"AIIA_TextDebugSplicer": "Text Debug Splicer (AIIA)"
|
|
}
|