102 lines
4.4 KiB
Python
102 lines
4.4 KiB
Python
import importlib
|
|
import logging
|
|
from pathlib import Path
|
|
from typing import Dict, Any, Set
|
|
import sys
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
NODE_CLASS_MAPPINGS: Dict[str, Any] = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS: Dict[str, str] = {}
|
|
|
|
IGNORED_MODULES: Set[str] = {'__init__.py'}
|
|
|
|
REQUIRED_MODULES: Set[str] = set()
|
|
|
|
PACKAGE_DISPLAY_NAME = "ComfyUI_RaykoStudio"
|
|
|
|
def load_modules():
|
|
try:
|
|
current_dir = Path(__file__).parent
|
|
logger.debug(f"Сканируем директорию: {current_dir}")
|
|
|
|
loaded_modules = []
|
|
failed_modules = []
|
|
for py_file in current_dir.glob('*.py'):
|
|
filename = py_file.name
|
|
if filename in IGNORED_MODULES:
|
|
logger.debug(f"Пропускаем игнорируемый файл: {filename}")
|
|
continue
|
|
|
|
module_name = py_file.stem
|
|
logger.debug(f"Попытка загрузки модуля: {module_name}")
|
|
|
|
try:
|
|
module = importlib.import_module(f'.{module_name}', package=__name__)
|
|
_merge_module_mappings(module, module_name)
|
|
loaded_modules.append(module_name)
|
|
except ImportError as e:
|
|
failed_modules.append((module_name, str(e)))
|
|
logger.error(f"Ошибка импорта модуля {module_name}: {e}")
|
|
except Exception as e:
|
|
failed_modules.append((module_name, str(e)))
|
|
logger.error(f"Неожиданная ошибка в модуле {module_name}: {e}")
|
|
|
|
missing_required = REQUIRED_MODULES - set(loaded_modules)
|
|
if missing_required:
|
|
logger.error(f"Обязательные модули не загружены: {missing_required}")
|
|
|
|
if not failed_modules:
|
|
logger.info(f"Все модули {PACKAGE_DISPLAY_NAME} успешно загружены.")
|
|
else:
|
|
for mod_name, error in failed_modules:
|
|
logger.error(f"Модуль '{mod_name}' не загружен: {error}")
|
|
logger.warning(f"Загружено {len(loaded_modules)} модулей, ошибки в {len(failed_modules)} модулях.")
|
|
|
|
except Exception as e:
|
|
logger.critical(f"Критическая ошибка при загрузке модулей: {e}")
|
|
raise
|
|
|
|
def _merge_module_mappings(module, module_name: str):
|
|
try:
|
|
if hasattr(module, 'NODE_CLASS_MAPPINGS'):
|
|
new_classes = module.NODE_CLASS_MAPPINGS
|
|
duplicates = set(NODE_CLASS_MAPPINGS.keys()) & set(new_classes.keys())
|
|
if duplicates:
|
|
logger.warning(f"Найдены дубликаты классов в модуле {module_name}: {duplicates}")
|
|
NODE_CLASS_MAPPINGS.update(new_classes)
|
|
logger.debug(f"Добавлены классы из {module_name}: {list(new_classes.keys())}")
|
|
|
|
if hasattr(module, 'NODE_DISPLAY_NAME_MAPPINGS'):
|
|
new_names = module.NODE_DISPLAY_NAME_MAPPINGS
|
|
duplicates = set(NODE_DISPLAY_NAME_MAPPINGS.keys()) & set(new_names.keys())
|
|
if duplicates:
|
|
logger.warning(f"Найдены дубликаты имен в модуле {module_name}: {duplicates}")
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(new_names)
|
|
logger.debug(f"Добавлены имена из {module_name}: {list(new_names.keys())}")
|
|
|
|
except Exception as e:
|
|
logger.error(f"Ошибка при объединении маппингов из модуля {module_name}: {e}")
|
|
raise
|
|
|
|
try:
|
|
load_modules()
|
|
except Exception as e:
|
|
logger.critical(f"Фатальная ошибка при инициализации пакета: {e}")
|
|
pass
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|
|
|
|
if not NODE_CLASS_MAPPINGS:
|
|
logger.warning("NODE_CLASS_MAPPINGS пуст - возможно, не найдено модулей с узлами")
|
|
else:
|
|
logger.debug(f"Загружено {len(NODE_CLASS_MAPPINGS)} классов узлов")
|
|
|
|
if not NODE_DISPLAY_NAME_MAPPINGS:
|
|
logger.debug("NODE_DISPLAY_NAME_MAPPINGS пуст - будут использованы имена классов по умолчанию")
|
|
else:
|
|
logger.debug(f"Загружено {len(NODE_DISPLAY_NAME_MAPPINGS)} отображаемых имен")
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS', 'WEB_DIRECTORY']
|