Spaces:
Running
Running
File size: 1,225 Bytes
583c1c7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
"""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): # pylint: disable = invalid-name, missing-function-docstring
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)
|