| """The Context Switch (Big)."""
|
| from .constants import get_category, get_name
|
| from .context_utils import (ORIG_CTX_RETURN_TYPES, ORIG_CTX_RETURN_NAMES, merge_new_context,
|
| get_orig_context_return_tuple, is_context_empty)
|
| from .utils import FlexibleOptionalInputType
|
|
|
|
|
| class RgthreeContextMerge:
|
| """The Context Merge node."""
|
|
|
| NAME = get_name("Context Merge")
|
| 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 = "merge"
|
|
|
| def get_return_tuple(self, ctx):
|
| """Returns the context data. Separated so it can be overridden."""
|
| return get_orig_context_return_tuple(ctx)
|
|
|
| def merge(self, **kwargs):
|
| """Merges any non-null passed contexts; later ones overriding earlier."""
|
| ctxs = [
|
| value for key, value in kwargs.items()
|
| if key.startswith('ctx_') and not is_context_empty(value)
|
| ]
|
| ctx = merge_new_context(*ctxs)
|
|
|
| return self.get_return_tuple(ctx)
|
|
|
