by-aeandCursor Agent
6b459d3211
Wrap imports in try-except blocks to provide clear error messages: - Node imports: log helpful message if ComfyUI dependencies missing - PromptServer import: log warning if server unavailable - Export empty mappings instead of crashing ComfyUI startup This allows the extension to fail gracefully when dependencies are unavailable (e.g., during testing or accidental standalone execution) while still providing diagnostic information. Co-authored-by: Cursor Agent <cursoragent@cursor.com>
49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
import logging
|
|
|
|
_logger = logging.getLogger(__name__)
|
|
|
|
try:
|
|
from .nodes.ideogram4_nodes import (
|
|
Ideogram4PromptBuilderKJEnhancedByAE,
|
|
Ideogram4PromptBuilderKJPreviewUpdateByAE,
|
|
)
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"Ideogram4PromptBuilderKJEnhancedByAE": Ideogram4PromptBuilderKJEnhancedByAE,
|
|
"Ideogram4PromptBuilderKJPreviewUpdateByAE": Ideogram4PromptBuilderKJPreviewUpdateByAE,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"Ideogram4PromptBuilderKJEnhancedByAE": "Ideogram 4 Prompt Builder Pro (based on the KJ node)",
|
|
"Ideogram4PromptBuilderKJPreviewUpdateByAE": "Ideogram 4 Prompt Builder Pro - Preview BG Update",
|
|
}
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|
|
|
|
except ImportError as e:
|
|
_logger.error(
|
|
"ComfyUI-ByAE-IdeogramTools: Failed to import nodes. "
|
|
"Make sure you are running this inside ComfyUI with required dependencies: %s", e
|
|
)
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
WEB_DIRECTORY = None
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|
|
|
|
try:
|
|
from server import PromptServer
|
|
from .routes import register_routes
|
|
|
|
if hasattr(PromptServer, "instance") and PromptServer.instance is not None:
|
|
try:
|
|
register_routes(PromptServer.instance.app)
|
|
except Exception:
|
|
_logger.exception("ComfyUI-ByAE-IdeogramTools: failed to register store routes")
|
|
except ImportError:
|
|
_logger.warning(
|
|
"ComfyUI-ByAE-IdeogramTools: PromptServer not available, "
|
|
"store routes will not be registered"
|
|
)
|