88 lines
3.0 KiB
Python
88 lines
3.0 KiB
Python
import os
|
|
import glob
|
|
import folder_paths
|
|
import importlib
|
|
from pathlib import Path
|
|
|
|
node_list = [
|
|
"server",
|
|
"api",
|
|
"easyNodes",
|
|
"image",
|
|
"logic"
|
|
]
|
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
|
|
for module_name in node_list:
|
|
imported_module = importlib.import_module(".py.{}".format(module_name), __name__)
|
|
NODE_CLASS_MAPPINGS = {**NODE_CLASS_MAPPINGS, **imported_module.NODE_CLASS_MAPPINGS}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {**NODE_DISPLAY_NAME_MAPPINGS, **imported_module.NODE_DISPLAY_NAME_MAPPINGS}
|
|
|
|
cwd_path = os.path.dirname(os.path.realpath(__file__))
|
|
comfy_path = folder_paths.base_path
|
|
|
|
#Wildcards读取
|
|
from .py.wildcards import read_wildcard_dict
|
|
wildcards_path = os.path.join(os.path.dirname(__file__), "wildcards")
|
|
if os.path.exists(wildcards_path):
|
|
read_wildcard_dict(wildcards_path)
|
|
else:
|
|
os.mkdir(wildcards_path)
|
|
|
|
#Styles
|
|
styles_path = os.path.join(os.path.dirname(__file__), "styles")
|
|
samples_path = os.path.join(os.path.dirname(__file__), "styles", "samples")
|
|
if os.path.exists(styles_path):
|
|
if not os.path.exists(samples_path):
|
|
os.mkdir(samples_path)
|
|
else:
|
|
os.mkdir(styles_path)
|
|
os.mkdir(samples_path)
|
|
|
|
#合并autocomplete覆盖到pyssss包
|
|
pyssss_path = os.path.join(comfy_path, "custom_nodes", "ComfyUI-Custom-Scripts", "user")
|
|
combine_folder = os.path.join(cwd_path, "autocomplete")
|
|
if os.path.exists(combine_folder):
|
|
pass
|
|
else:
|
|
os.mkdir(combine_folder)
|
|
if os.path.exists(pyssss_path):
|
|
output_file = os.path.join(pyssss_path, "autocomplete.txt")
|
|
# 遍历 combine 目录下的所有 txt 文件，读取内容并合并
|
|
merged_content = ''
|
|
for file_path in glob.glob(os.path.join(combine_folder, '*.txt')):
|
|
with open(file_path, 'r', encoding='utf-8', errors='ignore') as file:
|
|
try:
|
|
file_content = file.read()
|
|
merged_content += file_content + '\n'
|
|
except UnicodeDecodeError:
|
|
pass
|
|
# 备份之前的autocomplete
|
|
# bak_file = os.path.join(pyssss_path, "autocomplete.txt.bak")
|
|
# if os.path.exists(bak_file):
|
|
# pass
|
|
# elif os.path.exists(output_file):
|
|
# shutil.copy(output_file, bak_file)
|
|
if merged_content != '':
|
|
# 将合并的内容写入目标文件 autocomplete.txt，并指定编码为 utf-8
|
|
with open(output_file, 'w', encoding='utf-8') as target_file:
|
|
target_file.write(merged_content)
|
|
|
|
# ComfyUI-Easy-PS相关 (需要把模型预览图暴露给PS读取，此处借鉴了 AIGODLIKE-ComfyUI-Studio 的部分代码)
|
|
from .py.libs.add_resources import add_static_resource
|
|
from .py.libs.model import easyModelManager
|
|
model_config = easyModelManager().models_config
|
|
for model in model_config:
|
|
paths = folder_paths.get_folder_paths(model)
|
|
for path in paths:
|
|
if not Path(path).exists():
|
|
continue
|
|
add_static_resource(path, path, limit=True)
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS', "WEB_DIRECTORY"]
|
|
|
|
|
|
print('\033[34mComfy-Easy-Use (v1.1.4): \033[92mLoaded\033[0m') |