FilesExpand file tree

main

/

string_function.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

49 lines (44 loc) · 1.56 KB

main

/

string_function.py

Copy path

Top

File metadata and controls

• Code
• Blame

49 lines (44 loc) · 1.56 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

import re

class StringFunction:

@classmethod

def INPUT_TYPES(s):

return {

"required": {

"action": (["append", "replace"], {}),

"tidy_tags": (["yes", "no"], {}),

},

"optional": {

"text_a": ("STRING", {"multiline": True, "dynamicPrompts": False}),

"text_b": ("STRING", {"multiline": True, "dynamicPrompts": False}),

"text_c": ("STRING", {"multiline": True, "dynamicPrompts": False})

}

}

RETURN_TYPES = ("STRING",)

FUNCTION = "exec"

CATEGORY = "utils"

OUTPUT_NODE = True

def exec(self, action, tidy_tags, text_a="", text_b="", text_c=""):

tidy_tags = tidy_tags == "yes"

out = ""

if action == "append":

out = (", " if tidy_tags else "").join(filter(None, [text_a, text_b, text_c]))

else:

if text_c is None:

text_c = ""

if text_b.startswith("/") and text_b.endswith("/"):

regex = text_b[1:-1]

out = re.sub(regex, text_c, text_a)

else:

out = text_a.replace(text_b, text_c)

if tidy_tags:

out = re.sub(r"\s{2,}", " ", out)

out = out.replace(" ,", ",")

out = re.sub(r",{2,}", ",", out)

out = out.strip()

return {"ui": {"text": (out,)}, "result": (out,)}

NODE_CLASS_MAPPINGS = {

"StringFunction|pysssss": StringFunction,

}

NODE_DISPLAY_NAME_MAPPINGS = {

"StringFunction|pysssss": "String Function 🐍",

}