Spaces:
Running
on
L40S
Running
on
L40S
File size: 1,800 Bytes
4450790 |
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 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
"""The Dynamic Context node."""
from mimetypes import add_type
from .constants import get_category, get_name
from .utils import ByPassTypeTuple, FlexibleOptionalInputType
class RgthreeDynamicContext:
"""The Dynamic Context node.
Similar to the static Context and Context Big nodes, this allows users to add any number and
variety of inputs to a Dynamic Context node, and return the outputs by key name.
"""
NAME = get_name("Dynamic Context")
CATEGORY = get_category()
@classmethod
def INPUT_TYPES(cls): # pylint: disable = invalid-name,missing-function-docstring
return {
"required": {},
"optional": FlexibleOptionalInputType(add_type),
"hidden": {},
}
RETURN_TYPES = ByPassTypeTuple(("RGTHREE_DYNAMIC_CONTEXT",))
RETURN_NAMES = ByPassTypeTuple(("CONTEXT",))
FUNCTION = "main"
def main(self, **kwargs):
"""Creates a new context from the provided data, with an optional base ctx to start.
This node takes a list of named inputs that are the named keys (with an optional "+ " prefix)
which are to be stored within the ctx dict as well as a list of keys contained in `output_keys`
to determine the list of output data.
"""
base_ctx = kwargs.get('base_ctx', None)
output_keys = kwargs.get('output_keys', None)
new_ctx = base_ctx.copy() if base_ctx is not None else {}
for key_raw, value in kwargs.items():
if key_raw in ['base_ctx', 'output_keys']:
continue
key = key_raw.upper()
if key.startswith('+ '):
key = key[2:]
new_ctx[key] = value
print(new_ctx)
res = [new_ctx]
output_keys = output_keys.split(',') if output_keys is not None else []
for key in output_keys:
res.append(new_ctx[key] if key in new_ctx else None)
return tuple(res)
|