mirror of
https://github.com/justUmen/Bjornulf_custom_nodes.git
synced 2026-03-21 12:42:11 -03:00
130 lines
5.7 KiB
Python
130 lines
5.7 KiB
Python
import re
|
|
|
|
class TextReplace:
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"input_text": ("STRING", {"multiline": True, "forceInput": True}),
|
|
"search_text": ("STRING", {"multiline": True}),
|
|
"replace_text": ("STRING", {"multiline": True, "default": ""}),
|
|
"replace_count": ("INT", {"default": 0, "min": 0, "max": 1000,
|
|
"display": "number",
|
|
"tooltip": "Number of replacements (0 = replace all)"}),
|
|
"use_regex": ("BOOLEAN", {"default": False}),
|
|
"case_sensitive": ("BOOLEAN", {"default": True, "tooltip": "Whether the search should be case-sensitive"}),
|
|
"trim_whitespace": (["none", "left", "right", "both"], {
|
|
"default": "none",
|
|
"tooltip": "Remove whitespace around the found text"
|
|
})
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("STRING",)
|
|
FUNCTION = "replace_text"
|
|
CATEGORY = "Bjornulf"
|
|
|
|
def replace_text(self, input_text, search_text, replace_text, replace_count, use_regex, case_sensitive, trim_whitespace):
|
|
try:
|
|
# Convert input to string
|
|
input_text = str(input_text)
|
|
|
|
# Prepare regex flags
|
|
regex_flags = 0
|
|
if not case_sensitive:
|
|
regex_flags |= re.IGNORECASE
|
|
|
|
# Debug print
|
|
# print(f"Input: {input_text}")
|
|
# print(f"Search Text: {search_text}")
|
|
# print(f"Replace Text: {replace_text}")
|
|
# print(f"Use Regex: {use_regex}")
|
|
# print(f"Regex Flags: {regex_flags}")
|
|
|
|
if use_regex:
|
|
# Ensure regex pattern is valid
|
|
try:
|
|
# Compile the regex pattern first
|
|
pattern = re.compile(search_text, flags=regex_flags)
|
|
|
|
# Perform replacement
|
|
if replace_count == 0:
|
|
# Replace all instances
|
|
result = pattern.sub(replace_text, input_text)
|
|
else:
|
|
# Replace specific number of instances
|
|
result = pattern.sub(replace_text, input_text, count=replace_count)
|
|
|
|
# Debug print
|
|
# print(f"Regex Result: {result}")
|
|
|
|
return (result,)
|
|
|
|
except re.error as regex_compile_error:
|
|
# print(f"Invalid Regex Pattern: {regex_compile_error}")
|
|
return (input_text,)
|
|
|
|
else:
|
|
# Standard string replacement
|
|
if not case_sensitive:
|
|
# Case-insensitive string replacement
|
|
result = input_text
|
|
count = 0
|
|
while search_text.lower() in result.lower() and (replace_count == 0 or count < replace_count):
|
|
# Find the index of the match
|
|
idx = result.lower().index(search_text.lower())
|
|
|
|
# Determine left and right parts
|
|
left_part = result[:idx]
|
|
right_part = result[idx + len(search_text):]
|
|
|
|
# Trim whitespace based on option
|
|
if trim_whitespace == "left":
|
|
left_part = left_part.rstrip()
|
|
elif trim_whitespace == "right":
|
|
right_part = right_part.lstrip()
|
|
elif trim_whitespace == "both":
|
|
left_part = left_part.rstrip()
|
|
right_part = right_part.lstrip()
|
|
|
|
# Reconstruct the string
|
|
result = left_part + replace_text + right_part
|
|
count += 1
|
|
else:
|
|
# Case-sensitive replacement
|
|
result = input_text
|
|
count = 0
|
|
while search_text in result and (replace_count == 0 or count < replace_count):
|
|
# Find the index of the match
|
|
idx = result.index(search_text)
|
|
|
|
# Determine left and right parts
|
|
left_part = result[:idx]
|
|
right_part = result[idx + len(search_text):]
|
|
|
|
# Trim whitespace based on option
|
|
if trim_whitespace == "left":
|
|
left_part = left_part.rstrip()
|
|
elif trim_whitespace == "right":
|
|
right_part = right_part.lstrip()
|
|
elif trim_whitespace == "both":
|
|
left_part = left_part.rstrip()
|
|
right_part = right_part.lstrip()
|
|
|
|
# Reconstruct the string
|
|
result = left_part + replace_text + right_part
|
|
count += 1
|
|
|
|
return (result,)
|
|
|
|
except Exception as e:
|
|
# print(f"Unexpected error during text replacement: {e}")
|
|
return (input_text,)
|
|
|
|
@classmethod
|
|
def IS_CHANGED(cls, input_text, search_text, replace_text, replace_count, use_regex, case_sensitive, trim_whitespace):
|
|
# Return float("NaN") to ensure the node always processes
|
|
return float("NaN") |