14 lines
319 B
Python
14 lines
319 B
Python
from .nodes import NODE_CLASS_MAPPINGS as NCM
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
**NCM,
|
|
}
|
|
|
|
def remove_cm_prefix(node_mapping: str) -> str:
|
|
if node_mapping.startswith("TC_"):
|
|
return node_mapping[3:]
|
|
return node_mapping
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {key: remove_cm_prefix(key) for key in NODE_CLASS_MAPPINGS}
|