add Replace Strings node

This commit is contained in:
GTSuya Studio
2023-12-01 20:34:06 -05:00
parent 5239619359
commit 1ee8bb6f0f
2 changed files with 29 additions and 4 deletions
+1
View File
@@ -12,6 +12,7 @@ NODE_CLASS_MAPPINGS = {
"Wildcards": Wildcards,
"Danbooru (Random)": DanbooruRandom,
"Danbooru (ID)": DanbooruID,
"Replace Strings": ReplaceStrings,
}
__all__ = ['NODE_CLASS_MAPPINGS']
+28 -4
View File
@@ -56,7 +56,7 @@ class SimpleWildcards:
print(myline)
text = text.replace('__'+wildcard+'__',myline)
return (text,seed)
return (text,seed,)
#---------------------------------------------------------------------------------------------------------------------------------------------------#
@@ -89,7 +89,7 @@ class SimpleWildcardsDir:
print(myline)
text = text.replace('__'+wildcard+'__',myline)
return (text,seed)
return (text,seed,)
#---------------------------------------------------------------------------------------------------------------------------------------------------#
@@ -165,7 +165,7 @@ class DanbooruRandom:
pass
if tags and img_url :
break
return (tags,img_url)
return (tags,img_url,)
#---------------------------------------------------------------------------------------------------------------------------------------------------#
@@ -198,10 +198,33 @@ class DanbooruID:
pass
if tags and img_url :
break
return (tags,img_url)
return (tags,img_url,)
#---------------------------------------------------------------------------------------------------------------------------------------------------#
class ReplaceStrings:
@classmethod
def INPUT_TYPES(s):
return {
"required": {
"text": ("STRING", {"default": ""}),
"replace_list": ("STRING", {"default": ""}),
},
}
RETURN_TYPES = ("STRING",)
RETURN_NAMES = ("text",)
FUNCTION = "get_value"
CATEGORY = "GtsuyaStudio/Tools"
def get_value(self, text, replace_list):
for multi_words in replace_list.splitlines():
words = multi_words.split("|")
text = text.replace(words[0],words[1])
text = ' '.join(text.split())
return (text,)
#---------------------------------------------------------------------------------------------------------------------------------------------------#
# A dictionary that contains the friendly/humanly readable titles for the nodes
NODE_DISPLAY_NAME_MAPPINGS = {
"SimpleWildcards": "Simple Wildcards",
@@ -209,4 +232,5 @@ NODE_DISPLAY_NAME_MAPPINGS = {
"Wildcards": "Wildcards",
"DanbooruRandom": "Danbooru (Random)",
"DanbooruID": "Danbooru (ID)",
"ReplaceStrings": "Replace Strings",
}