80 lines
2.6 KiB
Python
Executable File
80 lines
2.6 KiB
Python
Executable File
import re
|
|
|
|
class SearchBySubstring:
|
|
# Class method to define input types
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"text": ("STRING", {"multiline": True, "default": "whole_word", "tooltip": "Source to be searched"}),
|
|
"pattern": ("STRING", {"default": "eyes", "tooltip": "Substring to match (e.g., 'eyes')"}),
|
|
},
|
|
"optional": {
|
|
"default": ("STRING", {"default": "not found", "tooltip": "Fallback value if 'pattern' is not found"}),
|
|
}
|
|
}
|
|
|
|
# Define return types and names
|
|
RETURN_TYPES = ("STRING", "LIST", "INT", "BOOLEAN")
|
|
RETURN_NAMES = ("match", "matchList", "count", "found")
|
|
|
|
FUNCTION = "searchBySubstring"
|
|
CATEGORY = "Phando/Text"
|
|
|
|
def searchBySubstring(self, text: str, pattern: str, default="not found") -> tuple:
|
|
pattern = pattern.encode().decode("unicode_escape").lower()
|
|
text = text.lower()
|
|
|
|
regex = rf'\b\w*{pattern}\w*\b'
|
|
matchList = re.findall(regex, text)
|
|
|
|
match = matchList[0] if matchList else default
|
|
|
|
# Prepare the output tuple
|
|
output = (match, matchList, len(matchList), len(matchList) > 0)
|
|
return output
|
|
|
|
|
|
class TextConcatenateDynamic:
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
min_input = 2
|
|
max_input = 10
|
|
inputs = {
|
|
"required": {
|
|
"delimiter": ("STRING", {"default": ",", "tooltip": "Delimiter to use for concatenation"}),
|
|
"count": ("INT", {"default": min_input, "min": min_input, "max": max_input}),
|
|
},
|
|
}
|
|
|
|
inputCount = inputs['required']['count'][1]['default']
|
|
for i in range(1, inputCount+1):
|
|
inputs["dynamic"][f"text_{i+1}"] = ("STRING", {"default": "", "tooltip": f"Text input {i + 1}"})
|
|
|
|
return inputs
|
|
|
|
# Define the return types
|
|
RETURN_TYPES = ("STRING",)
|
|
RETURN_NAMES = ("output",)
|
|
|
|
FUNCTION = "textConcatenateDynamic"
|
|
CATEGORY = "Phando/Text"
|
|
|
|
def textConcatenateDynamic(self, delimiter, count, optional_lora_stack=None, **kwargs):
|
|
inputs = [kwargs.get(f"text_{i + 1}", "") for i in range(count)]
|
|
concatenated_result = delimiter.join(inputs)
|
|
return (concatenated_result,)
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
'SearchBySubstring': SearchBySubstring,
|
|
'TextConcatenateDynamic': TextConcatenateDynamic,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
'SearchBySubstring': 'Search by Substring',
|
|
'TextConcatenateDynamic': 'Text Concatenate - Dynamic',
|
|
} |