66 lines
1.7 KiB
Python
66 lines
1.7 KiB
Python
from .Colors import cprint, Colors as Colors
|
|
import os
|
|
py_name=os.path.basename(__file__)
|
|
cprint(py_name, Colors.BLUE)
|
|
|
|
import os
|
|
import sys
|
|
#print(f"sys.modules : {sys.modules}")
|
|
|
|
#mn=os.path.abspath(__main__)
|
|
#print("nm ", mn)
|
|
|
|
#filePath = __file__
|
|
#print("This script file path is ", filePath)
|
|
#
|
|
#absFilePath = os.path.abspath(__file__)
|
|
#print("This script absolute path is ", absFilePath)
|
|
#
|
|
#realFilePath = os.path.realpath(__file__)
|
|
#print("This script real path is ", realFilePath)
|
|
#
|
|
#path, filename = os.path.split(absFilePath)
|
|
#print("Script file path is {}, filename is {}".format(path, filename))
|
|
|
|
nm=os.path.abspath(__name__)
|
|
print("abspath __name__ : ", nm)
|
|
|
|
wd = os.getcwd()
|
|
print("working directory : ", wd)
|
|
|
|
md=nm.replace(wd+"\\","")
|
|
print("import name", md)
|
|
|
|
#if md in sys.modules:
|
|
# print(f"{md!r} already in sys.modules")
|
|
#else:
|
|
# print(f"{md!r} not in sys.modules")
|
|
|
|
if not md.startswith("custom_nodes."):
|
|
|
|
import importlib
|
|
#from custom_nodes.ComfyUI_node_Lilly import eval(f"{name}")
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
}
|
|
|
|
def add(name):
|
|
#print(f"Load : {name}")
|
|
try:
|
|
pkg = importlib.import_module(f"{md}.{name}")
|
|
NODE_CLASS_MAPPINGS[name]=eval(f"pkg.{name}")
|
|
cprint(f"Load ok : {name}")
|
|
except Exception as e:
|
|
cprint(f"Load fail : {name}",Colors.YELLOW)
|
|
cprint(f"Exception : {e}",Colors.YELLOW)
|
|
|
|
add("CheckpointLoaderRandom")
|
|
add("CheckpointLoaderSimpleText")
|
|
add("CLIPTextEncodeWildcards")
|
|
add("LoraLoaderText")
|
|
add("LoraLoaderTextRandom")
|
|
add("Random_Sampler")
|
|
add("VAELoaderDecode")
|
|
add("SimpleSampler")
|
|
add("SaveImageSimple")
|
|
#add("test") |