38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
import importlib.util
|
|
import glob
|
|
import os
|
|
import sys
|
|
from .lam import init, get_ext_dir
|
|
import time
|
|
from server import PromptServer
|
|
|
|
repo_dir = os.path.dirname(os.path.realpath(__file__))
|
|
sys.path.insert(0, repo_dir)
|
|
original_modules = sys.modules.copy()
|
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
|
|
if init():
|
|
print("Loading ComfyUI-Lam")
|
|
py = get_ext_dir("py")
|
|
files = os.listdir(py)
|
|
for file in files:
|
|
if not file.endswith(".py"):
|
|
continue
|
|
name = os.path.splitext(file)[0]
|
|
try:
|
|
imported_module = importlib.import_module(".py.{}".format(name), __name__)
|
|
NODE_CLASS_MAPPINGS = {**NODE_CLASS_MAPPINGS, **imported_module.NODE_CLASS_MAPPINGS}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {**NODE_DISPLAY_NAME_MAPPINGS, **imported_module.NODE_DISPLAY_NAME_MAPPINGS}
|
|
except Exception as e:
|
|
print("节点：'"+name+"'导入异常",e)
|
|
|
|
WEB_DIRECTORY = "./js"
|
|
|
|
file_directory = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
PromptServer.instance.app.router.add_static("/wechatauth/static", file_directory+"/pages/static")
|
|
PromptServer.instance.app.router.add_static("/paint-board", file_directory+"/pages/paint-board")
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS","WEB_DIRECTORY"]
|