130 lines
5.2 KiB
Python
130 lines
5.2 KiB
Python
"""
|
||
LLS ComfyUI Custom Nodes — 插件入口
|
||
====================================
|
||
|
||
目录结构
|
||
--------
|
||
node/
|
||
├── __init__.py ← 本文件：自动扫描并汇总所有子包的注册表
|
||
├── nodes.py ← 共享工具函数（不含节点类）
|
||
├── upscale/ ← 图像超分（已实现：LLSUpscaleSwitcher）
|
||
├── sampling/ ← 采样与去噪
|
||
├── latent/ ← Latent 空间操作
|
||
├── image/ ← 图像处理与后处理
|
||
├── mask/ ← 遮罩操作
|
||
├── conditioning/ ← 文本编码与条件生成
|
||
├── model_loader/ ← 模型加载与管理
|
||
├── controlnet/ ← ControlNet / 控制引导
|
||
├── lora/ ← LoRA / Adapter
|
||
├── video/ ← 视频生成与帧处理
|
||
├── audio/ ← 音频处理
|
||
└── utils/ ← 数据类型与逻辑工具
|
||
|
||
扩展方式
|
||
--------
|
||
1. 在对应子包的 nodes.py 中添加节点类
|
||
2. 将类名注册到该文件的 NODE_CLASS_MAPPINGS / NODE_DISPLAY_NAME_MAPPINGS
|
||
3. 无需修改本文件，下次 ComfyUI 重启时自动生效
|
||
|
||
新增功能域
|
||
----------
|
||
在 node/ 下新建目录并包含 __init__.py（暴露两个映射字典）即可，
|
||
无需修改本文件——动态扫描会自动发现。
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import importlib
|
||
import types
|
||
|
||
WEB_DIRECTORY = "./web"
|
||
|
||
# ---------- 子包白名单（控制加载顺序，同时作为文档） ----------
|
||
# 按功能分类总览章节顺序排列；新增子包追加到列表末尾即可。
|
||
_SUBPACKAGES: list[str] = [
|
||
"model_loader", # 第 1 节：模型加载与管理
|
||
"conditioning", # 第 2 节：文本编码与条件生成
|
||
"sampling", # 第 3 节：图像采样与去噪
|
||
"qwen", # 第 4 节：Qwen 一体化节点
|
||
"flux2klein", # 第 4 节（Flux2Klein 子域）：编辑文本编码
|
||
"latent", # 第 4 节：Latent 空间操作
|
||
"image", # 第 5 节：图像处理与后处理
|
||
"repair", # 第 5 节（修复子域）：图像修复
|
||
"mask_draw", # 第 5 节（修复子域）：交互式遮罩绘制
|
||
"pro_edit", # 第 5 节（专业编辑子域）：图像编辑/重绘
|
||
"upscale", # 第 5 节（超分子域）：图像超分
|
||
"mask", # 第 6 节：遮罩操作
|
||
"controlnet", # 第 7 节：ControlNet / 控制引导
|
||
"lora", # 第 8 节：LoRA / Adapter
|
||
"video", # 第 10 节：视频生成与帧处理
|
||
"audio", # 第 11 节：音频处理
|
||
"utils", # 第 14 节：数据类型与逻辑工具
|
||
]
|
||
|
||
|
||
def _merge_subpackage(pkg_name: str, classes: dict, display_names: dict) -> None:
|
||
"""
|
||
动态导入子包并将其注册表合并到传入的字典中。
|
||
|
||
冲突检测：若同名节点已被前面的子包注册，发出警告后跳过，
|
||
保持先注册优先（first-wins）语义，避免静默覆盖。
|
||
"""
|
||
full_name = f"{__name__}.{pkg_name}"
|
||
try:
|
||
mod: types.ModuleType = importlib.import_module(full_name)
|
||
except Exception as exc: # pragma: no cover
|
||
# 子包导入失败不应阻断整个插件加载
|
||
print(f"[LLS] WARNING: Failed to import subpackage '{pkg_name}': {exc}")
|
||
return
|
||
|
||
sub_classes: dict = getattr(mod, "NODE_CLASS_MAPPINGS", {})
|
||
sub_names: dict = getattr(mod, "NODE_DISPLAY_NAME_MAPPINGS", {})
|
||
|
||
for key, cls in sub_classes.items():
|
||
if key in classes:
|
||
existing_mod = getattr(classes[key], "__module__", "unknown")
|
||
print(
|
||
f"[LLS] WARNING: Node key '{key}' from '{pkg_name}' conflicts with "
|
||
f"existing registration from '{existing_mod}'. Skipping."
|
||
)
|
||
continue
|
||
classes[key] = cls
|
||
if key in sub_names:
|
||
display_names[key] = sub_names[key]
|
||
|
||
|
||
def _merge_root_nodes(classes: dict, display_names: dict) -> None:
|
||
"""合并根级 nodes.py 中定义的注册表。"""
|
||
try:
|
||
mod: types.ModuleType = importlib.import_module(f"{__name__}.nodes")
|
||
except Exception as exc: # pragma: no cover
|
||
print(f"[LLS] WARNING: Failed to import root nodes module: {exc}")
|
||
return
|
||
|
||
root_classes: dict = getattr(mod, "NODE_CLASS_MAPPINGS", {})
|
||
root_names: dict = getattr(mod, "NODE_DISPLAY_NAME_MAPPINGS", {})
|
||
|
||
for key, cls in root_classes.items():
|
||
if key in classes:
|
||
existing_mod = getattr(classes[key], "__module__", "unknown")
|
||
print(
|
||
f"[LLS] WARNING: Root node key '{key}' conflicts with existing registration "
|
||
f"from '{existing_mod}'. Skipping."
|
||
)
|
||
continue
|
||
classes[key] = cls
|
||
if key in root_names:
|
||
display_names[key] = root_names[key]
|
||
|
||
|
||
# ---------- 构建全局注册表 ----------
|
||
|
||
NODE_CLASS_MAPPINGS: dict[str, type] = {}
|
||
NODE_DISPLAY_NAME_MAPPINGS: dict[str, str] = {}
|
||
|
||
_merge_root_nodes(NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS)
|
||
|
||
for _pkg in _SUBPACKAGES:
|
||
_merge_subpackage(_pkg, NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS)
|
||
|
||
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|