Initial commit

Adds text formatter.
2024-11-22 00:16:28 +01:00
commit 63f4709f84
6 changed files with 363 additions and 0 deletions
@@ -0,0 +1,129 @@
+from typing import Any
+
+CATEGORY_NAME = "exectails"
+
+
+class AnyType(str):
+    def __ne__(self, __value: object) -> bool:
+        return False
+
+
+any = AnyType("*")
+
+
+class ETTextFormatterNode:
+    """
+    A node that takes a number of text inputs and concatenates them into
+    a single string.
+    """
+
+    INPUT_IS_LIST = True
+
+    RETURN_TYPES = ("STRING",)
+    RETURN_NAMES = ("text",)
+
+    CATEGORY = CATEGORY_NAME
+    FUNCTION = "process"
+
+    def process(self, format, arg0="", arg1="", arg2="", arg3="", arg4="", arg5="", arg6="", arg7="", arg8="", arg9="") -> tuple:
+        # This looks pretty stupid, but it makes sense. I promise.
+        # Technically, we just want to accept strings that we can
+        # format, but due to how ComfyUI works, lists of strings
+        # can be plugged into string inputs as if they were plain
+        # strings.
+        #
+        # What happens then is that the system takes the length of
+        # the longest list passed in, padding the other ones to the
+        # same length, and calling the function for each entry. These
+        # entries then make up the output list.
+        #
+        # To avoid this, you can set the input to be a list, process
+        # will only be called once with all lists, and we can control
+        # what the output will actually be. The caveat is that we need
+        # do handle the lists somehow, so we flatten them to one string.
+        # Easy!
+
+        format = self.flatten(format)
+        arg0 = self.flatten(arg0)
+        arg1 = self.flatten(arg1)
+        arg2 = self.flatten(arg2)
+        arg3 = self.flatten(arg3)
+        arg4 = self.flatten(arg4)
+        arg5 = self.flatten(arg5)
+        arg6 = self.flatten(arg6)
+        arg7 = self.flatten(arg7)
+        arg8 = self.flatten(arg8)
+        arg9 = self.flatten(arg9)
+
+        result = format.format(arg0=arg0, arg1=arg1, arg2=arg2, arg3=arg3, arg4=arg4, arg5=arg5, arg6=arg6, arg7=arg7, arg8=arg8, arg9=arg9)
+
+        return (result,)
+
+    def flatten(self, value: list) -> str:
+        return " ".join([str(s) for s in value])
+
+
+class ETTextFormatter2Node(ETTextFormatterNode):
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "format": ("STRING", {"multiline": True}),
+            },
+            "optional": {
+                "arg0": ("STRING", {"forceInput": True}),
+                "arg1": ("STRING", {"forceInput": True}),
+            }
+        }
+
+
+class ETTextFormatter5Node(ETTextFormatterNode):
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "format": ("STRING", {"multiline": True}),
+            },
+            "optional": {
+                "arg0": ("STRING", {"forceInput": True}),
+                "arg1": ("STRING", {"forceInput": True}),
+                "arg2": ("STRING", {"forceInput": True}),
+                "arg3": ("STRING", {"forceInput": True}),
+                "arg4": ("STRING", {"forceInput": True}),
+            }
+        }
+
+
+class ETTextFormatter10Node(ETTextFormatterNode):
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "format": ("STRING", {"multiline": True}),
+            },
+            "optional": {
+                "arg0": ("STRING", {"forceInput": True}),
+                "arg1": ("STRING", {"forceInput": True}),
+                "arg2": ("STRING", {"forceInput": True}),
+                "arg3": ("STRING", {"forceInput": True}),
+                "arg4": ("STRING", {"forceInput": True}),
+                "arg5": ("STRING", {"forceInput": True}),
+                "arg6": ("STRING", {"forceInput": True}),
+                "arg7": ("STRING", {"forceInput": True}),
+                "arg8": ("STRING", {"forceInput": True}),
+                "arg9": ("STRING", {"forceInput": True}),
+            }
+        }
+
+
+NODE_CLASS_MAPPINGS = {
+    "ETTextFormatter2Node": ETTextFormatter2Node,
+    "ETTextFormatter5Node": ETTextFormatter5Node,
+    "ETTextFormatter10Node": ETTextFormatter10Node,
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "ETTextFormatter2Node": "Text Formatter (2 Arguments)",
+    "ETTextFormatter5Node": "Text Formatter (5 Arguments)",
+    "ETTextFormatter10Node": "Text Formatter (10 Arguments)",
+}