28 lines
1.1 KiB
Python
28 lines
1.1 KiB
Python
from .common import FlexibleOptionalInputType, AnyType
|
|
|
|
optional = FlexibleOptionalInputType(AnyType('STRING'))
|
|
optional.update({
|
|
'previous': ('STRING', {'forceInput': True, 'default': '', 'multiline': True})
|
|
})
|
|
|
|
class StringToggle:
|
|
@classmethod
|
|
def INPUT_TYPES(cls) -> dict:
|
|
return {
|
|
'required': {
|
|
'separator': ('STRING', {'default': ', '}),
|
|
'reverse': ('BOOLEAN', {'default': False})
|
|
},
|
|
'optional': optional
|
|
}
|
|
RETURN_TYPES = ('STRING',)
|
|
FUNCTION = 'join'
|
|
CATEGORY = 'utils'
|
|
def join(self, separator: str, reverse: bool, previous: str = '', **kwargs) -> tuple[str]:
|
|
arguments = {}
|
|
for (argtype, unique), value in [(k.split('_'),v) for k,v in kwargs.items() if '_' in k]:
|
|
arguments.setdefault(argtype, {})[unique] = value
|
|
strings = ([previous] if previous else []) + [text for unique, text in arguments['text'].items() if arguments['append'][unique]]
|
|
if reverse: strings = strings[::-1]
|
|
return (separator.join(strings),)
|