48 lines
1.2 KiB
Python
48 lines
1.2 KiB
Python
import os
|
|
import shutil
|
|
|
|
|
|
from modules.launch_util import git_clone
|
|
|
|
|
|
def onerror(func, path, exc_info):
|
|
import stat
|
|
if not os.access(path, os.W_OK):
|
|
os.chmod(path, stat.S_IWUSR)
|
|
func(path)
|
|
else:
|
|
raise 'Failed to invoke "shutil.rmtree", git management failed.'
|
|
|
|
|
|
def get_empty_folder(path):
|
|
if os.path.isdir(path) or os.path.exists(path):
|
|
shutil.rmtree(path, onerror=onerror)
|
|
os.makedirs(path, exist_ok=True)
|
|
return path
|
|
|
|
|
|
comfy_repo = "https://github.com/comfyanonymous/ComfyUI"
|
|
comfy_commit_hash = None
|
|
|
|
comfy_temp_path = get_empty_folder(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'backend', 'comfy_temp'))
|
|
comfy_core_path = get_empty_folder(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'backend', 'headless'))
|
|
|
|
git_clone(comfy_repo, comfy_temp_path, "ComfyUI", comfy_commit_hash)
|
|
|
|
|
|
def get_item(name):
|
|
shutil.move(os.path.join(comfy_temp_path, name), os.path.join(comfy_core_path, name))
|
|
|
|
|
|
get_item('comfy')
|
|
get_item('comfy_extras')
|
|
get_item('custom_nodes')
|
|
get_item('latent_preview.py')
|
|
get_item('folder_paths.py')
|
|
get_item('nodes.py')
|
|
get_item('LICENSE')
|
|
|
|
shutil.rmtree(comfy_temp_path, onerror=onerror)
|
|
|
|
print('Backend is built.')
|