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

Add the ability to specify commit ID /tag or branch when installing nodes

Browse Source
This commit is contained in:
chenyijian 2024-12-24 19:07:27 +08:00
parent 171496c2ca
commit af114ecccc
2 changed files with 30 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
cm-cli.py
View File

@ -200,15 +200,22 @@ class Ctx:
cm_ctx = Ctx()
def parse_node(node: str):
if '@@' in node:
name, commit = node.split('@@', 1)
else:
name, commit = node, None
return name, commit
def install_node(node_name, is_all=False, cnt_msg=''):
node_name, commit_id = parse_node(node_name)
if core.is_valid_url(node_name):
# install via urls
res = core.gitclone_install([node_name])
res = core.gitclone_install([node_name],commits = [commit_id])
if not res:
print(f"[bold red]ERROR: An error occurred while installing '{node_name}'.[/bold red]")
else:
print(f"{cnt_msg} [INSTALLED] {node_name:50}")
print(f"{cnt_msg} [INSTALLED] {node_name:50} => {commit_id}")
else:
node_path, node_item = cm_ctx.lookup_node_path(node_name)
@ -218,11 +225,11 @@ def install_node(node_name, is_all=False, cnt_msg=''):
elif os.path.exists(node_path + '.disabled'):
enable_node(node_name)
else:
res = core.gitclone_install(node_item['files'], instant_execution=True, msg_prefix=f"[{cnt_msg}] ")
res = core.gitclone_install(node_item['files'], instant_execution=True, msg_prefix=f"[{cnt_msg}] ", commits=[commit_id])
if not res:
print(f"[bold red]ERROR: An error occurred while installing '{node_name}'.[/bold red]")
else:
print(f"{cnt_msg} [INSTALLED] {node_name:50}")
print(f"{cnt_msg} [INSTALLED] {node_name:50} => {commit_id}")
def reinstall_node(node_name, is_all=False, cnt_msg=''):

22
glob/manager_core.py
View File

@ -584,9 +584,16 @@ def is_valid_url(url):
return False
def gitclone_install(files, instant_execution=False, msg_prefix=''):
print(f"{msg_prefix}Install: {files}")
for url in files:
def gitclone_install(files, instant_execution=False, msg_prefix='', commits=None):
print(f"{msg_prefix}Install: {files}:{commits}")
if commits is None:
commits = [None] * len(files)
if len(files) != len(commits):
print("Error: The number of files and commit IDs must match.")
return False
for url, commit_id in zip(files, commits):
if not is_valid_url(url):
print(f"Invalid git url: '{url}'")
return False
@ -605,6 +612,15 @@ def gitclone_install(files, instant_execution=False, msg_prefix=''):
return False
else:
repo = git.Repo.clone_from(url, repo_path, recursive=True, progress=GitProgress())
if commit_id:
print(f"Checkout commit: {commit_id}")
try:
# Try checking out as a commit, branch, or tag
repo.git.checkout(commit_id)
except Exception as checkout_error:
print(f"Error checking out {commit_id}: {checkout_error}")
return False
repo.git.clear_cache()
repo.close()