14 lines
707 B
Python
14 lines
707 B
Python
from .nodes.DOC_SaveImageAndAddToHistory import DOC_SaveImageAndAddToHistory
|
|
from .nodes.DOC_RandomPromptChoice import DOC_RandomPromptChoice
|
|
|
|
|
|
# A dictionary that contains all nodes you want to export with their names
|
|
# NOTE: names should be globally unique
|
|
NODE_CLASS_MAPPINGS = {
|
|
"DOC_SaveImageAndAddToHistory": DOC_SaveImageAndAddToHistory,
|
|
"DOC_RandomPromptChoice": DOC_RandomPromptChoice,
|
|
}
|
|
|
|
# This magic will use a property DISPLAY_NAME on each node to get the display name of the node for the UI
|
|
# This removees the need to define display names for all nodes in this file.
|
|
NODE_DISPLAY_NAME_MAPPINGS = {key: getattr(value, 'DISPLAY_NAME', None) for key, value in NODE_CLASS_MAPPINGS.items()} |