60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
"""The original Context Switch."""
|
|
from .constants import get_category, get_name
|
|
from .context_utils import is_context_empty
|
|
from .utils import ByPassTypeTuple
|
|
|
|
|
|
class RgthreeContextDynamicSwitch:
|
|
"""The initial Context Switch node.
|
|
|
|
For now, this will remain as-is but is otherwise backwards compatible with other Context nodes
|
|
outputs.
|
|
"""
|
|
|
|
NAME = get_name("Dynamic Context Switch")
|
|
CATEGORY = get_category()
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls): # pylint: disable = invalid-name, missing-function-docstring
|
|
return {
|
|
"required": {},
|
|
"optional": {
|
|
"output_keys": ("DYNAMIC_CONTEXT_OUTPUTS",), # This is a hidden widget of the output keys
|
|
"ctx_01": ("DYNAMIC_CONTEXT",),
|
|
"ctx_02": ("DYNAMIC_CONTEXT",),
|
|
"ctx_03": ("DYNAMIC_CONTEXT",),
|
|
"ctx_04": ("DYNAMIC_CONTEXT",),
|
|
"ctx_05": ("DYNAMIC_CONTEXT",),
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ByPassTypeTuple(("DYNAMIC_CONTEXT", ))
|
|
RETURN_NAMES = ByPassTypeTuple(("CONTEXT", ))
|
|
FUNCTION = "switch"
|
|
|
|
def switch(self, output_keys=None, ctx_01=None, ctx_02=None, ctx_03=None, ctx_04=None, ctx_05=None):
|
|
"""Chooses the first non-empty Context to output.
|
|
"""
|
|
base_ctx = None
|
|
if not is_context_empty(ctx_01):
|
|
base_ctx = ctx_01
|
|
elif not is_context_empty(ctx_02):
|
|
base_ctx = ctx_02
|
|
elif not is_context_empty(ctx_03):
|
|
base_ctx = ctx_03
|
|
elif not is_context_empty(ctx_04):
|
|
base_ctx = ctx_04
|
|
elif not is_context_empty(ctx_05):
|
|
base_ctx = ctx_05
|
|
|
|
new_ctx = base_ctx.copy() if not is_context_empty(base_ctx) else None
|
|
res = [new_ctx]
|
|
output_keys = output_keys.split(',') if output_keys is not None else []
|
|
for key in output_keys:
|
|
if new_ctx is None:
|
|
res.append(None)
|
|
else:
|
|
res.append(new_ctx[key] if key in new_ctx else None)
|
|
|
|
return tuple(res)
|