13 lines
441 B
Python
13 lines
441 B
Python
import os
|
|
import importlib
|
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
|
|
for node in os.listdir(os.path.dirname(__file__) + os.sep + 'nodes'):
|
|
if node.startswith('DT_'):
|
|
node = node.split('.')[0]
|
|
node_import = importlib.import_module('custom_nodes.ComfyUI-Vextra-Nodes.nodes.' + node)
|
|
print('Imported node: ' + node)
|
|
# get class node mappings from py file
|
|
NODE_CLASS_MAPPINGS.update(node_import.NODE_CLASS_MAPPINGS)
|