You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
112 lines
2.7 KiB
112 lines
2.7 KiB
import traceback |
|
|
|
# |
|
# Global Var |
|
# |
|
# Usage: |
|
# import cm_global |
|
# cm_global.variables['comfyui.revision'] = 1832 |
|
# print(f"log mode: {cm_global.variables['logger.enabled']}") |
|
# |
|
variables = {} |
|
|
|
|
|
# |
|
# Global API |
|
# |
|
# Usage: |
|
# [register API] |
|
# import cm_global |
|
# |
|
# def api_hello(msg): |
|
# print(f"hello: {msg}") |
|
# return msg |
|
# |
|
# cm_global.register_api('hello', api_hello) |
|
# |
|
# [use API] |
|
# import cm_global |
|
# |
|
# test = cm_global.try_call(api='hello', msg='an example') |
|
# print(f"'{test}' is returned") |
|
# |
|
|
|
APIs = {} |
|
|
|
|
|
def register_api(k, f): |
|
global APIs |
|
APIs[k] = f |
|
|
|
|
|
def try_call(**kwargs): |
|
if 'api' in kwargs: |
|
api_name = kwargs['api'] |
|
try: |
|
api = APIs.get(api_name) |
|
if api is not None: |
|
del kwargs['api'] |
|
return api(**kwargs) |
|
else: |
|
print(f"WARN: The '{kwargs['api']}' API has not been registered.") |
|
except Exception as e: |
|
print(f"ERROR: An exception occurred while calling the '{api_name}' API.") |
|
raise e |
|
else: |
|
return None |
|
|
|
|
|
# |
|
# Extension Info |
|
# |
|
# Usage: |
|
# import cm_global |
|
# |
|
# cm_global.extension_infos['my_extension'] = {'version': [0, 1], 'name': 'me', 'description': 'example extension', } |
|
# |
|
extension_infos = {} |
|
|
|
on_extension_registered_handlers = {} |
|
|
|
|
|
def register_extension(extension_name, v): |
|
global extension_infos |
|
global on_extension_registered_handlers |
|
extension_infos[extension_name] = v |
|
|
|
if extension_name in on_extension_registered_handlers: |
|
for k, f in on_extension_registered_handlers[extension_name]: |
|
try: |
|
f(extension_name, v) |
|
except Exception: |
|
print(f"[ERROR] '{k}' on_extension_registered_handlers") |
|
traceback.print_exc() |
|
|
|
del on_extension_registered_handlers[extension_name] |
|
|
|
|
|
def add_on_extension_registered(k, extension_name, f): |
|
global on_extension_registered_handlers |
|
if extension_name in extension_infos: |
|
try: |
|
v = extension_infos[extension_name] |
|
f(extension_name, v) |
|
except Exception: |
|
print(f"[ERROR] '{k}' on_extension_registered_handler") |
|
traceback.print_exc() |
|
else: |
|
if extension_name not in on_extension_registered_handlers: |
|
on_extension_registered_handlers[extension_name] = [] |
|
|
|
on_extension_registered_handlers[extension_name].append((k, f)) |
|
|
|
|
|
def add_on_revision_detected(k, f): |
|
if 'comfyui.revision' in variables: |
|
try: |
|
f(variables['comfyui.revision']) |
|
except Exception: |
|
print(f"[ERROR] '{k}' on_revision_detected_handler") |
|
traceback.print_exc() |
|
else: |
|
variables['cm.on_revision_detected_handler'].append((k, f))
|
|
|