75 lines
1.9 KiB
Python
75 lines
1.9 KiB
Python
# encoding: utf-8
|
|
"""
|
|
"""
|
|
|
|
import typing as _t
|
|
|
|
from inspect import cleandoc as _cleandoc
|
|
|
|
from frozendict import deepfreeze as _deepfreeze
|
|
|
|
from comfy.comfy_types.node_typing import IO as _IO
|
|
|
|
from . import _meta
|
|
from .docstring_formatter import format_docstring as _format_docstring
|
|
from .enums import DataTypes as _DataTypes
|
|
from .funcs_common import _show_text_on_node, _new_dict_with_updated_key, _T
|
|
|
|
|
|
_input_types = _deepfreeze({
|
|
'required': {
|
|
'name': (_IO.STRING, {'tooltip': (
|
|
"Name (key) of a string to extract from dictionary.\n"
|
|
"If the element with such key isn't a string, it will be turned to one.\n"
|
|
"If no such key exists in the dict, an empty string returned."
|
|
)}),
|
|
'show_status': (_IO.BOOLEAN, {'default': False, 'label_on': 'value', 'label_off': 'no', 'tooltip': (
|
|
"Show the extracted string on the node itself?"
|
|
)}),
|
|
},
|
|
'optional': {
|
|
'dict': _DataTypes.input_dict(tooltip="The dictionary to extract the element from."),
|
|
},
|
|
'hidden': {
|
|
'unique_id': 'UNIQUE_ID', # used for text display at the bottom of the node
|
|
},
|
|
})
|
|
|
|
|
|
class StringConstructorDictExtractString:
|
|
"""Extract a single string from the Format-Dict."""
|
|
NODE_NAME = 'StringConstructorDictExtractString'
|
|
CATEGORY = _meta.category_dict
|
|
DESCRIPTION = _format_docstring(_cleandoc(__doc__))
|
|
|
|
FUNCTION = 'main'
|
|
RETURN_TYPES = (_IO.STRING, )
|
|
RETURN_NAMES = ('string', )
|
|
# OUTPUT_TOOLTIPS = tuple()
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return _input_types
|
|
|
|
@staticmethod
|
|
def main(
|
|
name: str, show_status: bool = False, dict: _t.Dict[str, _T] = None,
|
|
unique_id: str = None,
|
|
) -> _t.Tuple[str]:
|
|
"""Extract a single string from the Format-Dict."""
|
|
# noinspection PyBroadException
|
|
try:
|
|
string: str = dict.get(name, '')
|
|
except Exception:
|
|
string = ''
|
|
if string is None:
|
|
string = ''
|
|
|
|
if not isinstance(string, str):
|
|
string = repr(string)
|
|
|
|
if show_status and unique_id:
|
|
_show_text_on_node(string, unique_id)
|
|
|
|
return (string, )
|