98 lines
3.8 KiB
Python
98 lines
3.8 KiB
Python
import importlib
|
|
import os
|
|
import shutil
|
|
import sys
|
|
import time
|
|
import json
|
|
|
|
WEB_DIRECTORY = os.path.join(os.path.dirname(__file__), "web")
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
|
|
|
|
def _load_locale_display_names():
|
|
"""Load zh locale display names for logging purposes."""
|
|
locale_map = {}
|
|
try:
|
|
base_dir = os.path.dirname(__file__)
|
|
locale_path = os.path.join(base_dir, "locales", "zh", "nodeDefs.json")
|
|
if os.path.exists(locale_path):
|
|
with open(locale_path, "r", encoding="utf-8") as f:
|
|
data = json.load(f)
|
|
for key, cfg in data.items():
|
|
disp = cfg.get("display_name")
|
|
if isinstance(disp, str) and disp:
|
|
# 去掉末尾装饰符（如 ☀）以及多余空格
|
|
clean = disp.rstrip()
|
|
if clean.endswith("☀"):
|
|
clean = clean[:-1].rstrip()
|
|
locale_map[key] = clean
|
|
except Exception:
|
|
# Logging is best-effort only; ignore locale errors
|
|
pass
|
|
return locale_map
|
|
|
|
|
|
def load_all_nodes(root_dir):
|
|
success_nodes = []
|
|
failed_nodes = []
|
|
|
|
zh_display_map = _load_locale_display_names()
|
|
|
|
for dirpath, _, filenames in os.walk(root_dir):
|
|
for filename in filenames:
|
|
if filename.endswith(".py") and filename != "__init__.py":
|
|
module_path = os.path.join(dirpath, filename)
|
|
rel_path = os.path.relpath(module_path, root_dir)
|
|
module_name = rel_path[:-3].replace(os.sep, ".")
|
|
try:
|
|
spec = importlib.util.spec_from_file_location(module_name, module_path)
|
|
module = importlib.util.module_from_spec(spec)
|
|
spec.loader.exec_module(module)
|
|
if hasattr(module, "NODE_CLASS_MAPPINGS"):
|
|
NODE_CLASS_MAPPINGS.update(module.NODE_CLASS_MAPPINGS)
|
|
display_map = getattr(module, "NODE_DISPLAY_NAME_MAPPINGS", {}) or {}
|
|
for name in module.NODE_CLASS_MAPPINGS.keys():
|
|
display_name = display_map.get(name, name)
|
|
# 去掉末尾的 ☀，用于英文内部名展示
|
|
if display_name and display_name.endswith("☀"):
|
|
display_name = display_name[:-1]
|
|
zh_name = zh_display_map.get(name, "")
|
|
success_nodes.append((name, display_name, zh_name))
|
|
if hasattr(module, "NODE_DISPLAY_NAME_MAPPINGS"):
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(module.NODE_DISPLAY_NAME_MAPPINGS)
|
|
except Exception as e:
|
|
# store as tuple (module_name, error_message) so callers can unpack
|
|
failed_nodes.append((module_name, str(e)))
|
|
|
|
separator = "=" * 64
|
|
print(separator)
|
|
print("🌞🌞🌞 DaShuai Tools 🌞🌞🌞")
|
|
print(f"✅ 已加载 {len(success_nodes)} 个节点:")
|
|
|
|
# 对齐英文列，让中文描述排版更整齐
|
|
max_eng_len = 0
|
|
for name, _, _ in success_nodes:
|
|
if len(str(name)) > max_eng_len:
|
|
max_eng_len = len(str(name))
|
|
|
|
for name, display_name, zh_name in success_nodes:
|
|
eng = name
|
|
cn = zh_name if zh_name else ""
|
|
eng_padded = eng.ljust(max_eng_len)
|
|
if cn and cn != eng:
|
|
# 例如：🌞 api_caption 🌞 API 打标
|
|
print(f" 🌞 {eng_padded} 🌞 {cn}")
|
|
else:
|
|
print(f" 🌞 {eng}")
|
|
if failed_nodes:
|
|
print(f"❌ 加载失败 {len(failed_nodes)} 个节点:")
|
|
for name, err in failed_nodes:
|
|
print(f" 🥹 {name}: {err}")
|
|
print(separator)
|
|
|
|
|
|
load_all_nodes(os.path.dirname(__file__))
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|