145 lines
5.3 KiB
Python
145 lines
5.3 KiB
Python
"""
|
|
██╗ ██████╗ ██╗ ██╗██╗ ███████╗██████╗ ██████╗ ██╗ ██╗████████╗
|
|
██║██╔═══██╗██║ ██║██║ ██╔════╝██╔══██╗██╔═══██╗██║ ██║╚══██╔══╝
|
|
██║██║ ██║██║ ██║██║ ███████╗██████╔╝██║ ██║██║ ██║ ██║
|
|
██ ██║██║ ██║╚██╗ ██╔╝██║ ╚════██║██╔═══╝ ██║ ██║██║ ██║ ██║
|
|
╚█████╔╝╚██████╔╝ ╚████╔╝ ██║ ███████║██║ ╚██████╔╝╚██████╔╝ ██║
|
|
╚════╝ ╚═════╝ ╚═══╝ ╚═╝ ╚══════╝╚═╝ ╚═════╝ ╚═════╝ ╚═╝
|
|
|
|
SPOUT support for ComfyUI
|
|
"""
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|
|
__author__ = "Alexander G. Morano"
|
|
__email__ = "amorano@gmail.com"
|
|
|
|
import os
|
|
import sys
|
|
import json
|
|
import inspect
|
|
import importlib
|
|
from pathlib import Path
|
|
from types import ModuleType
|
|
|
|
from loguru import logger
|
|
|
|
# ==============================================================================
|
|
# === GLOBAL ===
|
|
# ==============================================================================
|
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
ROOT = Path(__file__).resolve().parent
|
|
ROOT_COMFY = ROOT.parent.parent
|
|
|
|
JOV_WEB = ROOT / 'web'
|
|
|
|
JOV_LOG_LEVEL = os.getenv("JOV_LOG_LEVEL", "INFO")
|
|
logger.configure(handlers=[{"sink": sys.stdout, "level": JOV_LOG_LEVEL}])
|
|
|
|
JOV_INTERNAL = os.getenv("JOV_INTERNAL", 'false').strip().lower() in ('true', '1', 't')
|
|
|
|
JOV_PACKAGE = "JOV_SPOUT"
|
|
|
|
# ==============================================================================
|
|
# === SUPPORT ===
|
|
# ==============================================================================
|
|
|
|
def load_module(name: str) -> None|ModuleType:
|
|
module = inspect.getmodule(inspect.stack()[0][0]).__name__
|
|
try:
|
|
route = str(name).replace("\\", "/")
|
|
route = route.split(f"{module}/core/")[1]
|
|
route = route.split('.')[0].replace('/', '.')
|
|
except Exception as e:
|
|
logger.warning(f"module failed {name}")
|
|
logger.warning(str(e))
|
|
return
|
|
|
|
try:
|
|
module = f"{module}.core.{route}"
|
|
module = importlib.import_module(module)
|
|
except Exception as e:
|
|
logger.warning(f"module failed {module}")
|
|
logger.warning(str(e))
|
|
return
|
|
|
|
return module
|
|
|
|
def loader():
|
|
global NODE_DISPLAY_NAME_MAPPINGS, NODE_CLASS_MAPPINGS
|
|
NODE_LIST_MAP = {}
|
|
|
|
for fname in ROOT.glob('core/**/*.py'):
|
|
if fname.stem.startswith('_'):
|
|
continue
|
|
|
|
if (module := load_module(fname)) is None:
|
|
continue
|
|
|
|
classes = inspect.getmembers(module, inspect.isclass)
|
|
for class_name, class_object in classes:
|
|
if not class_name.endswith('BaseNode') and hasattr(class_object, 'NAME') and hasattr(class_object, 'CATEGORY'):
|
|
name = f"{class_object.NAME} ({JOV_PACKAGE})"
|
|
NODE_DISPLAY_NAME_MAPPINGS[name] = name
|
|
NODE_CLASS_MAPPINGS[name] = class_object
|
|
desc = class_object.DESCRIPTION if hasattr(class_object, 'DESCRIPTION') else name
|
|
NODE_LIST_MAP[name] = desc.split('.')[0].strip('\n')
|
|
|
|
NODE_CLASS_MAPPINGS = {x[0] : x[1] for x in sorted(NODE_CLASS_MAPPINGS.items(),
|
|
key=lambda item: getattr(item[1], 'SORT', 0))}
|
|
|
|
keys = NODE_CLASS_MAPPINGS.keys()
|
|
for name in keys:
|
|
logger.debug(f"✅ {name}")
|
|
logger.info(f"{len(keys)} nodes loaded")
|
|
|
|
# only do the list on local runs...
|
|
if JOV_INTERNAL:
|
|
with open(str(ROOT) + "/node_list.json", "w", encoding="utf-8") as f:
|
|
json.dump(NODE_LIST_MAP, f, sort_keys=True, indent=4 )
|
|
|
|
# ==============================================================================
|
|
# === CLASS ===
|
|
# ==============================================================================
|
|
|
|
class JOVBaseNode:
|
|
NOT_IDEMPOTENT = True
|
|
CATEGORY = f"{JOV_PACKAGE} 📺"
|
|
RETURN_TYPES = ()
|
|
FUNCTION = "run"
|
|
|
|
@classmethod
|
|
def IS_CHANGED(cls, **kw) -> float:
|
|
return float('nan')
|
|
|
|
@classmethod
|
|
def VALIDATE_INPUTS(cls, *arg, **kw) -> bool:
|
|
return True
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls, prompt:bool=False, extra_png:bool=False, dynprompt:bool=False) -> dict:
|
|
data = {
|
|
"optional": {},
|
|
"required": {},
|
|
"hidden": {
|
|
"ident": "UNIQUE_ID"
|
|
}
|
|
}
|
|
if prompt:
|
|
data["hidden"]["prompt"] = "PROMPT"
|
|
if extra_png:
|
|
data["hidden"]["extra_pnginfo"] = "EXTRA_PNGINFO"
|
|
|
|
if dynprompt:
|
|
data["hidden"]["dynprompt"] = "DYNPROMPT"
|
|
return data
|
|
|
|
# ==============================================================================
|
|
# === BOOTSTRAP ===
|
|
# ==============================================================================
|
|
|
|
loader()
|