TBNilles/ComfyUI-Manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: cm_api -> cm_global.APIs

Browse Source
This commit is contained in:
dr.lt.data
2024-01-02 17:06:59 +09:00
parent 7dfbdfb44e
commit 097a832e7e
4 changed files with 150 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
prestartup_script.py
View File

@@ -7,13 +7,18 @@ import threading
import re
import locale
import platform
import traceback
sys.CM_api = {}
glob_path = os.path.join(os.path.dirname(__file__), "glob")
sys.path.append(glob_path)
import cm_global
message_collapses = []
import_failed_extensions = set()
cm_global.variables['cm.on_revision_detected_handler'] = []
def register_message_collapse(f):
@@ -21,13 +26,16 @@ def register_message_collapse(f):
message_collapses.append(f)
def is_import_failed_extension(x):
def is_import_failed_extension(name):
global import_failed_extensions
return x in import_failed_extensions
return name in import_failed_extensions
sys.__comfyui_manager_register_message_collapse = register_message_collapse
sys.__comfyui_manager_is_import_failed_extension = is_import_failed_extension
cm_global.register_api('cm.register_message_collapse', register_message_collapse)
cm_global.register_api('cm.is_import_failed_extension', is_import_failed_extension)
comfyui_manager_path = os.path.dirname(__file__)
custom_nodes_path = os.path.abspath(os.path.join(comfyui_manager_path, ".."))
@@ -442,4 +450,4 @@ if os.path.exists(script_list_path):
print("#######################################################################\n")
del processed_install
del pip_list
del pip_list