60 lines
2.1 KiB
Python
60 lines
2.1 KiB
Python
class TextWithContext:
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
|
|
return {"required": {
|
|
"text": ("STRING", {"multiline": True, "default": ""})
|
|
}}
|
|
|
|
_all_context_input_output_data = {
|
|
"base_ctx": ("base_ctx", "RGTHREE_CONTEXT", "CONTEXT"),
|
|
"text": ("text", "STRING", "TEXT"),
|
|
}
|
|
|
|
_original_ctx_inputs_list = [
|
|
"base_ctx", "text"
|
|
]
|
|
|
|
RETURN_TYPES = ("RGTHREE_CONTEXT","STRING", )
|
|
FUNCTION = "convert"
|
|
CATEGORY = "SimpleOfComplex/TextContext"
|
|
OUTPUT_NODE = True
|
|
#RETURN_TYPES = ORIG_CTX_RETURN_TYPES
|
|
#RETURN_NAMES = ORIG_CTX_RETURN_NAMES
|
|
|
|
def convert(self, base_ctx=None, **kwargs): # pylint: disable = missing-function-docstring
|
|
ctx = self.new_context(base_ctx, **kwargs)
|
|
return self.get_orig_context_return_tuple(ctx)
|
|
|
|
def new_context(self, base_ctx, **kwargs):
|
|
"""Creates a new context from the provided data, with an optional base ctx to start."""
|
|
context = base_ctx if base_ctx is not None else None
|
|
new_ctx = {}
|
|
for key in self._all_context_input_output_data:
|
|
if key == "base_ctx":
|
|
continue
|
|
v = kwargs[key] if key in kwargs else None
|
|
new_ctx[key] = v if v is not None else context[
|
|
key] if context is not None and key in context else None
|
|
return new_ctx
|
|
|
|
def get_orig_context_return_tuple(self,ctx):
|
|
"""Returns a tuple for returning from a node with only the original context keys."""
|
|
return self.get_context_return_tuple(ctx, self._original_ctx_inputs_list)
|
|
|
|
def get_context_return_tuple(self,ctx, inputs_list=None):
|
|
"""Returns a tuple for returning in the order of the inputs list."""
|
|
if inputs_list is None:
|
|
inputs_list = self._all_context_input_output_data.keys()
|
|
tup_list = [
|
|
ctx,
|
|
]
|
|
for key in inputs_list:
|
|
if key == "base_ctx":
|
|
continue
|
|
tup_list.append(ctx[key] if ctx is not None and key in ctx else None)
|
|
return tuple(tup_list)
|
|
|
|
|