Add replace string, make globally unique.

2024-06-21 21:07:24 -04:00
parent 9f70d78db2
commit e32c99d508
3 changed files with 45 additions and 9 deletions
@@ -0,0 +1,33 @@
+import os
+import json
+
+class ReplaceString:
+    def __init__(self):
+        pass
+
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "haystack": ("STRING",{"defaultInput": True, "default": "{}", "multiline": True}),
+                "needle": ("STRING",{"defaultInput": False, "default": "%REPLACE%", "dynamicPrompts": False}),
+                "value": ("STRING",{"defaultInput": True, "default": "", "dynamicPrompts": False}),
+                "input_type": (["json", "verbatim"], {"default": "verbatim"}),
+            },
+            "optional": {
+            },
+        }
+
+    RETURN_TYPES = ("STRING", )
+    FUNCTION = "execute"
+
+    CATEGORY = "iw"
+
+    def execute(self, haystack, needle, value, input_type):
+        if input_type == "json":
+            z = json.loads(value)
+            if type(z) is str:
+                value = z
+            if type(z) is list:
+                value = " ".join(z)
+        return (haystack.replace(needle, value),)