xiaden
315666c7b6
- Split flat node modules into categorized subpackages (tagging, detect, edge, pose, segment, restore, compare, judge, transform, utility) - Extract shared internals into _shared/ (bases, tensor, formatting) - Centralize node registration via register_nodes() helper - Clean up: remove dead is_scored return, unnecessary hasattr guard, inline imports
58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
"""
|
|
imgutils_nodes — ComfyUI custom node pack wrapping deepghs/imgutils.
|
|
|
|
Provides 34 nodes for anime image analysis, tagging, detection, and processing.
|
|
Uses the ComfyUI V2 io.ComfyNode API.
|
|
"""
|
|
|
|
import logging
|
|
|
|
from comfy_api.latest import ComfyExtension, io
|
|
|
|
__version__ = "0.1.0"
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
|
|
_SUB_PACKAGES: list[str] = [
|
|
"tagging", "detect", "pose", "segment",
|
|
"edge", "restore", "compare", "judge",
|
|
"transform", "utility",
|
|
]
|
|
|
|
|
|
def _register_subpackage(pkg: str) -> None:
|
|
"""Import a subpackage and merge its node mappings into the root registries."""
|
|
try:
|
|
mod = __import__(
|
|
f"{__name__}.{pkg}",
|
|
fromlist=["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"],
|
|
)
|
|
NODE_CLASS_MAPPINGS.update(mod.NODE_CLASS_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(mod.NODE_DISPLAY_NAME_MAPPINGS)
|
|
except Exception:
|
|
logger.warning(f"{pkg.title()} nodes unavailable.", exc_info=True)
|
|
|
|
|
|
for _pkg in _SUB_PACKAGES:
|
|
_register_subpackage(_pkg)
|
|
|
|
n_nodes = len(NODE_CLASS_MAPPINGS)
|
|
logger.info(f"imgutils_nodes v{__version__}: Loaded {n_nodes} nodes.")
|
|
if n_nodes == 0:
|
|
logger.warning("imgutils_nodes: No nodes loaded. Check dghs-imgutils>=0.19.0.")
|
|
|
|
|
|
class ImgUtilsExtension(ComfyExtension):
|
|
async def get_node_list(self) -> list[type[io.ComfyNode]]:
|
|
return list(NODE_CLASS_MAPPINGS.values())
|
|
|
|
|
|
async def comfy_entrypoint() -> ImgUtilsExtension:
|
|
return ImgUtilsExtension()
|
|
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|