Ethanfel
b2f5850b46
Sidecar chains {content,name,ext} specs over a SIDECAR-typed link; the save node
mirrors SaveImageKJ (folder_paths.get_save_image_path) and writes each sidecar as
base+name+ext next to the image. build_plan validates the whole chain before any
I/O so duplicates/bad extensions write nothing.
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
40 lines
2.1 KiB
Python
40 lines
2.1 KiB
Python
"""ComfyUI-Datasete-Gates — custom nodes."""
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
# ComfyUI loads this directory as a package, so __package__ is set and the
|
|
# relative imports below resolve. pytest, however, collects the repo root as a
|
|
# Package and imports this file standalone (no parent package) during test
|
|
# setup — in that case the relative imports would raise. Guard on __package__
|
|
# so the test suite can import `gates.*` without dragging in aiohttp/comfy.
|
|
if __package__:
|
|
from .gates.node import NODE_CLASS_MAPPINGS as _POOL_NODES, \
|
|
NODE_DISPLAY_NAME_MAPPINGS as _POOL_NAMES
|
|
from .gates.loader import NODE_CLASS_MAPPINGS as _LOADER_NODES, \
|
|
NODE_DISPLAY_NAME_MAPPINGS as _LOADER_NAMES
|
|
from .gates.gate import NODE_CLASS_MAPPINGS as _GATE_NODES, \
|
|
NODE_DISPLAY_NAME_MAPPINGS as _GATE_NAMES
|
|
from .gates.textgate import NODE_CLASS_MAPPINGS as _TEXT_NODES, \
|
|
NODE_DISPLAY_NAME_MAPPINGS as _TEXT_NAMES
|
|
from .gates.profile_node import NODE_CLASS_MAPPINGS as _PROF_NODES, \
|
|
NODE_DISPLAY_NAME_MAPPINGS as _PROF_NAMES
|
|
from .gates.bucket_node import NODE_CLASS_MAPPINGS as _BUCKET_NODES, \
|
|
NODE_DISPLAY_NAME_MAPPINGS as _BUCKET_NAMES
|
|
from .gates.sidecar_node import NODE_CLASS_MAPPINGS as _SC_NODES, \
|
|
NODE_DISPLAY_NAME_MAPPINGS as _SC_NAMES
|
|
from .gates import routes # noqa: F401 (registers aiohttp routes on import)
|
|
from .gates import gate_server # noqa: F401 (registers /datasete_gate/* + text routes)
|
|
from .gates import profiles_routes # noqa: F401 (registers /grid_pool/profiles/*)
|
|
|
|
NODE_CLASS_MAPPINGS = {**_POOL_NODES, **_LOADER_NODES, **_GATE_NODES,
|
|
**_TEXT_NODES, **_PROF_NODES, **_BUCKET_NODES,
|
|
**_SC_NODES}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {**_POOL_NAMES, **_LOADER_NAMES, **_GATE_NAMES,
|
|
**_TEXT_NAMES, **_PROF_NAMES, **_BUCKET_NAMES,
|
|
**_SC_NAMES}
|
|
else: # pragma: no cover - exercised only under pytest collection
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|