| """The original Context Switch."""
|
| from .constants import get_category, get_name
|
| from .context_utils import (ORIG_CTX_RETURN_TYPES, ORIG_CTX_RETURN_NAMES, is_context_empty,
|
| get_orig_context_return_tuple)
|
| from .utils import FlexibleOptionalInputType
|
|
|
|
|
| class RgthreeContextSwitch:
|
| """The (original) Context Switch node."""
|
|
|
| NAME = get_name("Context Switch")
|
| CATEGORY = get_category()
|
|
|
| @classmethod
|
| def INPUT_TYPES(cls):
|
| return {
|
| "required": {},
|
| "optional": FlexibleOptionalInputType("RGTHREE_CONTEXT"),
|
| }
|
|
|
| RETURN_TYPES = ORIG_CTX_RETURN_TYPES
|
| RETURN_NAMES = ORIG_CTX_RETURN_NAMES
|
| FUNCTION = "switch"
|
|
|
| def get_return_tuple(self, ctx):
|
| """Returns the context data. Separated so it can be overridden."""
|
| return get_orig_context_return_tuple(ctx)
|
|
|
| def switch(self, **kwargs):
|
| """Chooses the first non-empty Context to output."""
|
| ctx = None
|
| for key, value in kwargs.items():
|
| if key.startswith('ctx_') and not is_context_empty(value):
|
| ctx = value
|
| break
|
| return self.get_return_tuple(ctx)
|
|
|
