File size: 1,098 Bytes
edb09f2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
class TextSplit:
    def __init__(self):
        pass

    @classmethod
    def INPUT_TYPES(s):
        return {
            "required": {
                "text": (
                    "STRING",
                    {"default": "", "forceInput": True},
                ),
                "separator": (
                    "STRING",
                    {"default": "."},
                ),
                "reverse": ([False, True],),
                "return_half": (["First Half", "Second Half"],),
            },
        }

    RETURN_TYPES = ("STRING",)
    RETURN_NAMES = ("text",)
    OUTPUT_NODE = True
    FUNCTION = "do_split"
    CATEGORY = "Chibi-Nodes/Text"

    def do_split(self, text, separator, reverse, return_half):
        if reverse:
            text = text.rsplit(separator, 1)
        else:
            text = text.split(separator, 1)

        if len(text) == 2:
            if return_half == "First Half":
                text = text[0]
            if return_half == "Second Half":
                text = text[1]

        return (text,)