wangbo/ComfyUI-Manager
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI-Manager.git synced 2026-02-14 23:22:35 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add 'exit-on-fail' parameter to control failure behavior

Browse Source
This commit is contained in:
chenyijian 2024-12-24 19:19:59 +08:00
parent af114ecccc
commit 3dbd0c324e
1 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
cm-cli.py
View File

@ -200,6 +200,7 @@ class Ctx:
cm_ctx = Ctx() cm_ctx = Ctx()
# use '@@' to separate node name and commit id
def parse_node(node: str): def parse_node(node: str):
if '@@' in node: if '@@' in node:
name, commit = node.split('@@', 1) name, commit = node.split('@@', 1)
@ -207,13 +208,17 @@ def parse_node(node: str):
name, commit = node, None name, commit = node, None
return name, commit return name, commit
def install_node(node_name, is_all=False, cnt_msg=''): def install_node(node_name, is_all=False, cnt_msg='', **kwargs):
node_name, commit_id = parse_node(node_name) node_name, commit_id = parse_node(node_name)
exit_on_fail = kwargs.get('exit_on_fail', False)
print(f"exit_on_fail:{exit_on_fail}...")
if core.is_valid_url(node_name): if core.is_valid_url(node_name):
# install via urls # install via urls
res = core.gitclone_install([node_name],commits = [commit_id]) res = core.gitclone_install([node_name],commits = [commit_id])
if not res: if not res:
print(f"[bold red]ERROR: An error occurred while installing '{node_name}'.[/bold red]") print(f"[bold red]ERROR: An error occurred while installing '{node_name}'.[/bold red]")
if exit_on_fail:
sys.exit(1)
else: else:
print(f"{cnt_msg} [INSTALLED] {node_name:50} => {commit_id}") print(f"{cnt_msg} [INSTALLED] {node_name:50} => {commit_id}")
else: else:
@ -228,6 +233,8 @@ def install_node(node_name, is_all=False, cnt_msg=''):
res = core.gitclone_install(node_item['files'], instant_execution=True, msg_prefix=f"[{cnt_msg}] ", commits=[commit_id]) res = core.gitclone_install(node_item['files'], instant_execution=True, msg_prefix=f"[{cnt_msg}] ", commits=[commit_id])
if not res: if not res:
print(f"[bold red]ERROR: An error occurred while installing '{node_name}'.[/bold red]") print(f"[bold red]ERROR: An error occurred while installing '{node_name}'.[/bold red]")
if exit_on_fail:
sys.exit(1)
else: else:
print(f"{cnt_msg} [INSTALLED] {node_name:50} => {commit_id}") print(f"{cnt_msg} [INSTALLED] {node_name:50} => {commit_id}")
@ -475,7 +482,7 @@ def auto_save_snapshot():
print(f"Current snapshot is saved as `{path}`") print(f"Current snapshot is saved as `{path}`")
def for_each_nodes(nodes, act, allow_all=True): def for_each_nodes(nodes, act, allow_all=True, **kwargs):
is_all = False is_all = False
if allow_all and 'all' in nodes: if allow_all and 'all' in nodes:
is_all = True is_all = True
@ -487,7 +494,7 @@ def for_each_nodes(nodes, act, allow_all=True):
i = 1 i = 1
for x in nodes: for x in nodes:
try: try:
act(x, is_all=is_all, cnt_msg=f'{i}/{total}') act(x, is_all=is_all, cnt_msg=f'{i}/{total}', **kwargs)
except Exception as e: except Exception as e:
print(f"ERROR: {e}") print(f"ERROR: {e}")
traceback.print_exc() traceback.print_exc()
@ -519,9 +526,13 @@ def install(
None, None,
help="[remote|local|cache]" help="[remote|local|cache]"
), ),
exit_on_fail: bool = typer.Option(
False,
help="Exit on failure"
)
): ):
cm_ctx.set_channel_mode(channel, mode) cm_ctx.set_channel_mode(channel, mode)
for_each_nodes(nodes, act=install_node) for_each_nodes(nodes, act=install_node, exit_on_fail=exit_on_fail)
@app.command(help="Reinstall custom nodes") @app.command(help="Reinstall custom nodes")